Searched defs:may_combine_partial_sharding (Results 1 – 2 of 2) sorted by relevance
77 bool may_combine_partial_sharding, in MaybeImproveInstructionSharding()389 bool may_combine_partial_sharding) { in InferDotShardingFromOperands()454 bool may_combine_partial_sharding) { in InferGatherParallelShardingFromOperands()534 bool may_combine_partial_sharding) { in InferConvolutionShardingFromOperands()640 int64_t operand_index, bool may_combine_partial_sharding) { in InferDotOperandSharding()755 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in InferShardingFromUsers() local1326 const bool may_combine_partial_sharding = is_spmd && aggressiveness > 0; in GetShardingFromUser() local1641 bool may_combine_partial_sharding) { in InferDynamicSliceOrDynamicUpdateSliceShardingFromOperands()1737 const bool may_combine_partial_sharding = is_spmd_ && aggressiveness > 0; in InferShardingFromOperands() local
79 bool may_combine_partial_sharding) { in MergeSharding()