Home
last modified time | relevance | path

Searched refs:reductionDimSize (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp98 int64_t reductionDimSize = loopRanges[reductionDim]; in splitReduction() local
99 if (reductionDimSize == ShapedType::kDynamicSize || in splitReduction()
100 reductionDimSize % ratio != 0 || in splitReduction()
291 int64_t reductionDimSize = loopRanges[reductionDimPos]; in splitReductionByScaling() local
292 if (reductionDimSize == ShapedType::kDynamicSize || in splitReductionByScaling()
293 reductionDimSize % splitFactor != 0 || in splitReductionByScaling()
338 reductionDimSize / splitFactor, insertSplitDimension); in splitReductionByScaling()
373 reductionDimSize / splitFactor)); in splitReductionByScaling()
381 loc, ArrayRef<int64_t>{reductionDimSize / splitFactor, splitFactor}, in splitReductionByScaling()