Searched defs:original_sharding (Results 1 – 1 of 1) sorted by relevance
907 HloSharding original_sharding = sharding(); in ReplicatePartial() local1014 auto original_sharding = partitioned_hlo.sharding(); in ReshardToPartialReplicateWithAllGather() local