collect_producer_nodes

class torch.ao.quantization.fx.utils.collect_producer_nodes(node) [source]

Начиная с целевого узла, выполняет трассировку назад до входного узла или узла getattr. Это используется для извлечения цепочки операторов, начинающейся с getattr и заканчивающейся целевым узлом, например:

def forward(self, x):
    observed = self.observer(self.weight)
    return F.linear(x, observed)

collect_producer_nodes(observed) вернёт либо список узлов, создающих узел observed, либо None, если не удаётся извлечь самодостаточный граф без свободных переменных (входов функции forward).

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

list[Node] | None