Home
last modified time | relevance | path

Searched refs:is_transparent_output_node (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/passes/infra/
H A Dpartitioner.py302 def is_transparent_output_node(node: Node, partition: Set[Node], removed_nodes: Set[Node]): function
309 if not is_transparent_output_node(output_n, partition, removed_nodes):
325 is_transparent_output_node(node, set(partition.nodes), remove_node)):