Searched defs:sharding_grouped (Results 1 – 3 of 3) sorted by relevance
581 auto sharding_grouped = GroupShardingOnDims( in HandleScatter() local
956 auto sharding_grouped = in ReshardToPartialReplicateWithAllGather() local3086 auto sharding_grouped = GroupShardingOnDims(hlo->sharding(), group_dims); in HandleRng() local
1551 auto sharding_grouped = GroupShardingOnDims(sharding, sharding_dims); in AlignShardingOnDims() local