Searched refs:dus_sharding (Results 1 – 1 of 1) sorted by relevance
2359 const auto& dus_sharding = hlo->sharding(); in HandleDynamicUpdateSlice() local2361 GetPartitionedHlo(hlo->operand(0)).Reshard(dus_sharding).hlo(); in HandleDynamicUpdateSlice()2368 hlo_sharding_util::PartiallyReplicateTiledShardingOnDims(dus_sharding, in HandleDynamicUpdateSlice()