Searched defs:may_combine_partial_sharding (Results 1 – 2 of 2) sorted by relevance
78 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() local1065 int64_t operand_index, bool may_combine_partial_sharding) { in InferDotOperandSharding()1149 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in GetShardingFromUser() local1451 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromUsers() local
70 bool may_combine_partial_sharding) { in MergeSharding()