Searched refs:find_all_gather_patterns (Results 1 – 2 of 2) sorted by relevance
11 find_all_gather_patterns,96 all_gathers = find_all_gather_patterns(gm.graph)
82 def find_all_gather_patterns(graph: torch.fx.Graph): function835 all_gathers = find_all_gather_patterns(graph)