Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h45 bool IsShardingMoreSpecific(const HloSharding& lhs, const HloSharding& rhs);
Dhlo_sharding_util.cc37 bool IsShardingMoreSpecific(const HloSharding& lhs, const HloSharding& rhs) { in IsShardingMoreSpecific() function
48 if (IsShardingMoreSpecific(rhs_shardings[i], lhs_shardings[i])) { in IsShardingMoreSpecific()
51 if (IsShardingMoreSpecific(lhs_shardings[i], rhs_shardings[i])) { in IsShardingMoreSpecific()
85 return IsShardingMoreSpecific(*to_merge, old); in MergeSharding()
109 return IsShardingMoreSpecific(*to_merge, old); in MergeSharding()
194 return IsShardingMoreSpecific(*to_merge, old); in MergeSharding()
Dsharding_propagation.cc221 hlo_sharding_util::IsShardingMoreSpecific( in PickRepresentativeOperand()
659 if (hlo_sharding_util::IsShardingMoreSpecific( in InferShardingFromOperands()
667 if (hlo_sharding_util::IsShardingMoreSpecific( in InferShardingFromOperands()