Searched refs:ReshardPartialReplicateWithAllToAll (Results 1 – 2 of 2) sorted by relevance
369 absl::optional<PartitionedHlo> ReshardPartialReplicateWithAllToAll(
423 try_reshard = ReshardPartialReplicateWithAllToAll(target); in ReshardNoCache()434 try_reshard = ReshardPartialReplicateWithAllToAll(target); in ReshardNoCache()1228 PartitionedHlo::ReshardPartialReplicateWithAllToAll(const HloSharding& target) { in ReshardPartialReplicateWithAllToAll() function in xla::spmd::PartitionedHlo