Home
last modified time | relevance | path

Searched refs:PartialReplicateReshardCompatibleSharding (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h384 absl::optional<HloSharding> PartialReplicateReshardCompatibleSharding(
Dspmd_partitioner.cc918 PartialReplicateReshardCompatibleSharding(target, sharding()); in ReshardToPartialReplicateWithAllGather()
989 PartialReplicateReshardCompatibleSharding(sharding(), target); in ReshardFromPartialReplicateWithDynamicSlice()
Dspmd_partitioner_util.cc303 absl::optional<HloSharding> PartialReplicateReshardCompatibleSharding( in PartialReplicateReshardCompatibleSharding() function