graph_module_from_producer_nodes#
- class torch.ao.quantization.fx.utils.graph_module_from_producer_nodes(root, producer_nodes)[source]#
Construct a graph module from extracted producer nodes from collect_producer_nodes function :param root: the root module for the original graph :param producer_nodes: a list of nodes we use to construct the graph
- Returns:
A graph module constructed from the producer nodes
- Return type:
