Searched refs:PartiallyReplicateTiledShardingOnDims (Results 1 – 5 of 5) sorted by relevance
106 if (hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in MaybeImproveInstructionSharding()362 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferDotShardingFromOperands()434 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferGatherParallelShardingFromOperands()718 : hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferShardingFromOperands()971 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferShardingFromOperands()1076 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferDotOperandSharding()1094 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in InferDotOperandSharding()1155 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in GetShardingFromUser()
189 HloSharding PartiallyReplicateTiledShardingOnDims(
981 HloSharding filtered_output_sharding = PartiallyReplicateTiledShardingOnDims( in GatherDataOperandShardingFromOutput()1142 HloSharding PartiallyReplicateTiledShardingOnDims( in PartiallyReplicateTiledShardingOnDims() function
1327 if (hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in ReshardWithCollectivePermute()1329 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in ReshardWithCollectivePermute()2252 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims(hlo->sharding(), in HandleBroadcast()2368 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims(dus_sharding, in HandleDynamicUpdateSlice()2801 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in HandleReduce()
2054 .Reshard(hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in PartitionDotGroupOnNonContracting()2216 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims( in PartitionDotGroupOnContracting()