Lines Matching defs:maskOp
262 vector::MaskOp maskOp, Attribute memorySpace, Operation *insertionPoint) {
263 assert(llvm::range_size(maskOp.getMaskBlock()->without_terminator()) == 1 &&
267 Operation *yieldOp = maskOp.getMaskRegion().front().getTerminator();
273 rewriter, options, maskOp.getMaskableOp(), memorySpace,
274 /*insertionPoint=*/insertionPoint ? insertionPoint : maskOp);
289 maskOp.walk([&](bufferization::ToTensorOp toTensorOp) {
298 for (Value result : maskOp.getResults())
302 rewriter.setInsertionPoint(maskOp);
303 if (failed(cast<bufferization::BufferizableOpInterface>(maskOp.getOperation())
478 if (auto maskOp = dyn_cast<vector::MaskOp>(op))
479 return bufferizeToAllocation(rewriter, options, maskOp, memorySpace);