Searched refs:find_reduce_scatter_patterns (Results 1 – 2 of 2) sorted by relevance
12 find_reduce_scatter_patterns,145 reduce_scatters = find_reduce_scatter_patterns(gm.graph)
218 def find_reduce_scatter_patterns(graph: torch.fx.Graph): function836 reduce_scatters = find_reduce_scatter_patterns(graph)