Searched defs:partitioned_hlo (Results 1 – 2 of 2) sorted by relevance
530 const PartitionedHlo& partitioned_hlo) { in SetPartitionedHlo()
1320 auto partitioned_hlo = *this; in ReshardToPartialReplicateWithAllGather() local1934 auto partitioned_hlo = in ReshardPartialReplicateWithAllToAll() local1939 auto partitioned_hlo = Reshard(tmp_partial_replicate_sharding); in ReshardPartialReplicateWithAllToAll() local