Searched refs:get_while_nodes (Results 1 – 1 of 1) sorted by relevance
650 def get_while_nodes(graph_module: torch.fx.GraphModule) -> Iterable[Node]: function774 for while_node in get_while_nodes(graph_module):