Home
last modified time | relevance | path

Searched defs:may_combine_partial_sharding (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc78 bool may_combine_partial_sharding) { in MaybeImproveInstructionSharding()
353 bool may_combine_partial_sharding) { in InferDotShardingFromOperands()
418 bool may_combine_partial_sharding) { in InferGatherParallelShardingFromOperands()
493 bool may_combine_partial_sharding) { in InferConvolutionShardingFromOperands()
618 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromOperands() local
1065 int64_t operand_index, bool may_combine_partial_sharding) { in InferDotOperandSharding()
1149 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in GetShardingFromUser() local
1451 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromUsers() local
Dhlo_sharding_util.cc70 bool may_combine_partial_sharding) { in MergeSharding()