Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h49 bool MergeSharding(const HloSharding& old, HloSharding* to_merge,
Dhlo_sharding_util.cc69 bool MergeSharding(const HloSharding& old, HloSharding* to_merge, in MergeSharding() function
76 MergeSharding(old.tuple_elements()[i], &to_merge->tuple_elements()[i], in MergeSharding()
996 if (MergeSharding(*parallel_sharding, &*passthrough_sharding, in GatherDataOperandShardingFromOutput()
1000 if (MergeSharding(*passthrough_sharding, &*parallel_sharding, in GatherDataOperandShardingFromOutput()
Dsharding_propagation.cc87 if (hlo_sharding_util::MergeSharding(instruction->sharding(), &sharding, in MaybeImproveInstructionSharding()
1017 hlo_sharding_util::MergeSharding(instruction->sharding(), &sharding, in InferShardingFromOperands()
1114 if (hlo_sharding_util::MergeSharding(sharding, &sharding_from_other, in InferDotOperandSharding()