Searched refs:GetPartitionGroupsForReplication (Results 1 – 1 of 1) sorted by relevance
192 std::vector<std::vector<int64>> GetPartitionGroupsForReplication( in GetPartitionGroupsForReplication() function3531 GetPartitionGroupsForReplication(sharding, {*it}); in AllGatherShardsInternal()3541 GetPartitionGroupsForReplication(sharding, selected_dims); in AllGatherShardsInternal()3615 GetPartitionGroupsForReplication(sharding, selected_dims); in AllReduceAlongShardingDimsInternal()3625 GetPartitionGroupsForReplication(sharding, {*it}); in AllReduceAlongShardingDimsInternal()