Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorMultiDimReductionTransforms.cpp123 SmallVector<bool> reductionMask = multiReductionOp.getReductionMask(); in matchAndRewrite()
H A DVectorTransforms.cpp986 SmallVector<bool> reductionMask = reduceOp.getReductionMask(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp186 static SmallVector<bool> getReductionMask(LinalgOp linalgOp) { in getReductionMask() function
317 SmallVector<bool> reductionMask = getReductionMask(linalgOp); in reduceIfNeeded()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/IR/
H A DVectorOps.td359 assert(d >= 0 && d < static_cast<int64_t>(getReductionMask().size()) &&
361 return getReductionMask()[d];
364 SmallVector<bool> getReductionMask() {
370 static SmallVector<bool> getReductionMask(