Searched defs:partitioned_hlo (Results 1 – 2 of 2) sorted by relevance
462 const PartitionedHlo& partitioned_hlo) { in SetPartitionedHlo()
924 auto partitioned_hlo = *this; in ReshardToPartialReplicateWithAllGather() local1297 auto partitioned_hlo = in ReshardPartialReplicateWithAllToAll() local1302 auto partitioned_hlo = Reshard(tmp_partial_replicate_sharding); in ReshardPartialReplicateWithAllToAll() local