评价此页

graph_module_from_producer_nodes#

class torch.ao.quantization.fx.utils.graph_module_from_producer_nodes(root, producer_nodes)[source]#

collect_producer_nodes 函数提取的生产者节点构造一个图模块 :param root: 原始图的根模块 :param producer_nodes: 用于构造图的节点列表

返回:

从生产者节点构造的图模块

返回类型:

GraphModule