Searched refs:reductionMask (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorMultiDimReductionTransforms.cpp | 82 SmallVector<bool> reductionMask(srcRank, false); in matchAndRewrite() local
85 reductionMask[srcRank - i - 1] = true; in matchAndRewrite()
87 reductionMask[i] = true; in matchAndRewrite()
91 reductionMask, multiReductionOp.getKind()); in matchAndRewrite()
123 SmallVector<bool> reductionMask = multiReductionOp.getReductionMask(); in matchAndRewrite() local
124 if (srcRank == 2 && reductionMask.front() != reductionMask.back()) in matchAndRewrite()
130 for (const auto &it : llvm::enumerate(reductionMask)) { in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 986 SmallVector<bool> reductionMask = reduceOp.getReductionMask(); in matchAndRewrite() local
987 auto srcMap = rewriter.getMultiDimIdentityMap(reductionMask.size()); in matchAndRewrite()
990 for (const auto &isReduceDim : llvm::enumerate(reductionMask)) { in matchAndRewrite()
998 auto dstMap = AffineMap::get(/*dimCount=*/reductionMask.size(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 179 const SmallVector<bool> &reductionMask) { in buildMultiDimReduce() argument
183 reduceOp->getLoc(), valueToReduce, acc, reductionMask, *maybeKind); in buildMultiDimReduce()
188 SmallVector<bool> reductionMask(linalgOp.iterator_types().size(), false); in getReductionMask() local
191 reductionMask[idx] = true; in getReductionMask()
194 return reductionMask; in getReductionMask()
317 SmallVector<bool> reductionMask = getReductionMask(linalgOp); in reduceIfNeeded() local
318 return buildMultiDimReduce(b, op, reduceVec, outputVec, reductionMask); in reduceIfNeeded()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 337 Value acc, ArrayRef<bool> reductionMask, in build() argument
340 for (const auto &en : llvm::enumerate(reductionMask)) in build()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/IR/ |
| H A D | VectorOps.td | 345 "ArrayRef<bool>":$reductionMask, "CombiningKind":$kind)>
|