Searched refs:make_all_gather_split_pattern (Results 1 – 1 of 1) sorted by relevance
99 def make_all_gather_split_pattern(shard): function120 make_all_gather_split_pattern(KeywordArg("shard")),140 make_all_gather_split_pattern(