Home
last modified time | relevance | path

Searched defs:reductionMask (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorMultiDimReductionTransforms.cpp82 SmallVector<bool> reductionMask(srcRank, false); in matchAndRewrite() local
123 SmallVector<bool> reductionMask = multiReductionOp.getReductionMask(); in matchAndRewrite() local
H A DVectorTransforms.cpp986 SmallVector<bool> reductionMask = reduceOp.getReductionMask(); in matchAndRewrite() local
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp179 const SmallVector<bool> &reductionMask) { in buildMultiDimReduce()
188 SmallVector<bool> reductionMask(linalgOp.iterator_types().size(), false); in getReductionMask() local
317 SmallVector<bool> reductionMask = getReductionMask(linalgOp); in reduceIfNeeded() local
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp337 Value acc, ArrayRef<bool> reductionMask, in build()