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