Searched defs:partitioned_hlo (Results 1 – 2 of 2) sorted by relevance
495 const PartitionedHlo& partitioned_hlo) { in SetPartitionedHlo()
974 auto partitioned_hlo = *this; in ReshardToPartialReplicateWithAllGather() local1347 auto partitioned_hlo = in ReshardPartialReplicateWithAllToAll() local1352 auto partitioned_hlo = Reshard(tmp_partial_replicate_sharding); in ReshardPartialReplicateWithAllToAll() local