Searched refs:original_sharding (Results 1 – 1 of 1) sorted by relevance
964 auto original_sharding = partitioned_hlo.sharding(); in ReshardToPartialReplicateWithAllGather() local970 partitioned_hlo.hlo()->set_sharding(original_sharding); in ReshardToPartialReplicateWithAllGather()