Function get_while_nodes

exir/memory_planning.py:1175–1178 · view source on GitHub ↗

(graph_module: torch.fx.GraphModule)

Source from the content-addressed store, hash-verified

1173
1174
1175	def get_while_nodes(graph_module: torch.fx.GraphModule) -> Iterable[Node]:
1176	for nd in graph_module.graph.nodes:
1177	if nd.target is exir_while:
1178	yield nd
1179
1180
1181	def get_map_nodes(graph_module: torch.fx.GraphModule) -> Iterable[Node]:

_apply_algo_to_submodulesFunction · 0.85

no outgoing calls

no test coverage detected