Function get_map_nodes

exir/memory_planning.py:1181–1184 · view source on GitHub ↗

(graph_module: torch.fx.GraphModule)

Source from the content-addressed store, hash-verified

1179
1180
1181	def get_map_nodes(graph_module: torch.fx.GraphModule) -> Iterable[Node]:
1182	for nd in graph_module.graph.nodes:
1183	if nd.target is torch.ops.higher_order.map_impl:
1184	yield nd
1185
1186
1187	def get_scan_nodes(graph_module: torch.fx.GraphModule) -> Iterable[Node]:

_apply_algo_to_submodulesFunction · 0.85

no outgoing calls

no test coverage detected