Searched defs:sharding_grouped (Results 1 – 4 of 4) sorted by relevance
110 auto sharding_grouped = in HandleCustomCallTopK() local
911 auto sharding_grouped = hlo_sharding_util::GroupShardingOnDims( in PartitionScatterIndexPassthroughPartition() local
1352 auto sharding_grouped = hlo_sharding_util::GroupShardingOnDims( in ReshardToPartialReplicateWithAllGather() local3877 auto sharding_grouped = in HandleRng() local
1434 auto sharding_grouped = in AlignShardingOnDims() local