Home
last modified time | relevance | path

Searched defs:target_sharding (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc1729 const HloSharding& target_sharding) { in SplitReshapeHelper()
1752 const HloSharding& target_sharding) { in MergeReshapeHelper()
H A Dspmd_partitioner_util.cc219 const HloSharding& partial_sharding, const HloSharding& target_sharding) { in PartialReplicateReshardCompatibleSharding()
H A Ddot_handler.cc2420 HloSharding target_sharding = UngroupSharding(*other_grouped); in PartitionDotGroupOnNonContracting() local