Home
last modified time | relevance | path

Searched defs:reductionDim (Results 1 – 5 of 5) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp48 unsigned reductionDim = dims[0]; in splitReduction() local
218 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim() local
232 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim() local
H A DTiling.cpp819 int reductionDim = static_cast<int>(redDims.front()); tileReductionUsingForall() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DLowerVectorScan.cpp127 int64_t reductionDim = scanOp.getReductionDim(); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1549 for (int64_t reductionDim : getDimensions()) getIteratorTypesArray() local
2710 int64_t reductionDim = getDimension(); decomposeOperation() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp6261 int64_t reductionDim = getReductionDim(); verify() local