graph_module_from_producer_nodes

class torch.ao.quantization.fx.utils.graph_module_from_producer_nodes(root, producer_nodes) [исходный код]

Создает модуль графа из извлеченных узлов-производителей функции collect_producer_nodes :param root: корневой модуль исходного графа :param producer_nodes: список узлов, используемых для создания графа

Возвращает:

Модуль графа, созданный из узлов-производителей

Тип возвращаемого значения:

GraphModule