Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp96 unsigned reductionDim = dims[0]; in splitReduction() local
98 int64_t reductionDimSize = loopRanges[reductionDim]; in splitReduction()
127 if (reductionDim == dim) { in splitReduction()
248 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim() local
255 reductionDim, reductionDim * reductionRatio + reductionDimP1, in scaleReductionDim()
262 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim() local
267 return map.compose(shiftedIdMap).insertResult(reductionDim, reductionDimPos); in insertParallelDim()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp2819 int64_t reductionDim = scanOp.getReductionDim(); in matchAndRewrite() local
2826 reductionShape[reductionDim] = 1; in matchAndRewrite()
2831 sizes[reductionDim] = 1; in matchAndRewrite()
2836 for (int i = 0; i < destShape[reductionDim]; i++) { in matchAndRewrite()
2837 offsets[reductionDim] = i; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp4843 int64_t reductionDim = getReductionDim(); in verify() local
4844 if (reductionDim >= srcRank) in verify()
4846 << reductionDim << " has to be less than " << srcRank; in verify()
4859 if (i != reductionDim) in verify()