Searched refs:may_combine_partial_sharding (Results 1 – 3 of 3) sorted by relevance
76 bool may_combine_partial_sharding) { in MaybeImproveInstructionSharding() argument88 may_combine_partial_sharding)) { in MaybeImproveInstructionSharding()339 bool may_combine_partial_sharding) { in InferDotShardingFromOperands() argument391 may_combine_partial_sharding); in InferDotShardingFromOperands()396 may_combine_partial_sharding); in InferDotShardingFromOperands()404 bool may_combine_partial_sharding) { in InferGatherParallelShardingFromOperands() argument464 instruction, may_combine_partial_sharding); in InferGatherParallelShardingFromOperands()471 instruction, may_combine_partial_sharding); in InferGatherParallelShardingFromOperands()479 bool may_combine_partial_sharding) { in InferConvolutionShardingFromOperands() argument515 may_combine_partial_sharding); in InferConvolutionShardingFromOperands()[all …]
50 bool may_combine_partial_sharding);
70 bool may_combine_partial_sharding) { in MergeSharding() argument77 may_combine_partial_sharding); in MergeSharding()81 if (!may_combine_partial_sharding || !old.ReplicateOnLastTileDim() || in MergeSharding()