Searched refs:ReshardToPartialReplicateWithAllGather (Results 1 – 2 of 2) sorted by relevance
361 absl::optional<PartitionedHlo> ReshardToPartialReplicateWithAllGather(
430 auto try_reshard = ReshardToPartialReplicateWithAllGather(target); in ReshardNoCache()910 PartitionedHlo::ReshardToPartialReplicateWithAllGather( in ReshardToPartialReplicateWithAllGather() function in xla::spmd::PartitionedHlo