Searched defs:may_combine_partial_sharding (Results 1 – 2 of 2) sorted by relevance
76 bool may_combine_partial_sharding) { in MaybeImproveInstructionSharding()339 bool may_combine_partial_sharding) { in InferDotShardingFromOperands()404 bool may_combine_partial_sharding) { in InferGatherParallelShardingFromOperands()479 bool may_combine_partial_sharding) { in InferConvolutionShardingFromOperands()601 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromOperands() local1048 int64 operand_index, bool may_combine_partial_sharding) { in InferDotOperandSharding()1132 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in GetShardingFromUser() local1432 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromUsers() local
70 bool may_combine_partial_sharding) { in MergeSharding()