Searched refs:make_zero_dim_all_gather_pattern (Results 1 – 1 of 1) sorted by relevance
85 def make_zero_dim_all_gather_pattern(shard): function97 zero_dim_all_gather_pattern = make_zero_dim_all_gather_pattern(KeywordArg("shard"))104 make_zero_dim_all_gather_pattern(shard),127 make_zero_dim_all_gather_pattern(