Lines Matching defs:opToMask

229   maskOperation(RewriterBase &rewriter, Operation *opToMask, LinalgOp linalgOp,
245   /// Create or retrieve an existing mask value to mask `opToMask` in the
249   Value getOrCreateMaskFor(RewriterBase &rewriter, Operation *opToMask,
382 /// Create or retrieve an existing mask value to mask `opToMask` in the
387     RewriterBase &rewriter, Operation *opToMask, LinalgOp linalgOp,
394   auto maskableOp = dyn_cast<vector::MaskableOpInterface>(opToMask);
456 VectorizationState::maskOperation(RewriterBase &rewriter, Operation *opToMask,
459   LDBG("Trying to mask: " << *opToMask << "\n");
467       getOrCreateMaskFor(rewriter, opToMask, linalgOp, maybeMaskingMap);
471     return opToMask;
475   assert(opToMask && "Expected a valid operation to mask");
477       mlir::vector::maskOperation(rewriter, opToMask, mask));
480   for (auto [resIdx, resVal] : llvm::enumerate(opToMask->getResults()))
3464                                Operation *opToMask) {
3466         return opToMask;
3471       auto xferOp = cast<VectorTransferOpInterface>(opToMask);
3476           cast<LinalgOp>(op).hasPureTensorSemantics(), opToMask, rewriter);
3481       return mlir::vector::maskOperation(rewriter, opToMask, maskOp);