torch.fx.experimental.merge_matmul.may_depend_on

torch.fx.experimental.merge_matmul.may_depend_on(a, b, search_depth=6) [исходный код]

Определяет, зависит ли один узел от другого в torch.fx.Graph.

Параметры:

• a (Node) – Узел, который может зависеть от b.
• b (Node) – Узел, от которого может зависеть a.
• search_depth (int) – В случае косвенной зависимости функция ищет зависимость данных на расстоянии до указанного количества узлов. Если такая зависимость не найдена, функция консервативно предполагает, что зависимость существует.

Возвращает:

True, если a может зависеть от b, и False, если такая зависимость точно отсутствует.

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

bool