Searched defs:partial_replicate_hlo (Results 1 – 1 of 1) sorted by relevance
915 auto partial_replicate_hlo = in ReplicatePartial() local1016 auto partial_replicate_hlo = PartitionedHlo(halo_exchange_hlo, base_shape, in ReshardToPartialReplicateWithAllGather() local