Lines Matching defs:maskableOp
681 auto maskableOp =
685 if (maskableOp.isMasked()) {
686 rewriter.setInsertionPoint(maskableOp.getMaskingOp());
687 rootOp = maskableOp.getMaskingOp();
688 mask = maskableOp.getMaskingOp().getMask();
6287 Operation *maskableOp,
6296 maskRegionBuilder(builder, maskableOp);
6301 Value mask, Operation *maskableOp,
6303 build(builder, result, resultTypes, mask, /*passthru=*/Value(), maskableOp,
6309 Value mask, Value passthru, Operation *maskableOp,
6311 build(builder, result, mask, maskableOp, maskRegionBuilder);
6434 auto maskableOp = dyn_cast<MaskableOpInterface>(block.front());
6435 if (!maskableOp)
6439 if (maskableOp->getNumResults() != getNumResults())
6443 if (!llvm::equal(maskableOp->getResultTypes(), getResultTypes()))
6447 if (llvm::count_if(maskableOp->getResultTypes(),
6452 Type expectedMaskType = maskableOp.getExpectedMaskType();
6460 if (!maskableOp.supportsPassthru())
6464 if (maskableOp->getNumResults() != 1)
6467 if (passthru.getType() != maskableOp->getResultTypes()[0])
6485 Operation *maskableOp = getMaskableOp();
6486 maskableOp->dropAllUses();
6487 maskableOp->moveBefore(getOperation());
6489 llvm::append_range(results, maskableOp->getResults());
6690 /// Create the vector.yield-ended region of a vector.mask op with `maskableOp`
6693 Operation *maskableOp) {
6694 assert(maskableOp->getBlock() && "MaskableOp must be inserted into a block");
6698 insBlock->begin(), maskableOp->getBlock()->getOperations(), maskableOp);
6699 builder.create<YieldOp>(maskableOp->getLoc(), maskableOp->getResults());
6706 Operation *maskableOp, Value mask,
6709 return maskableOp;
6711 return builder.create<MaskOp>(maskableOp->getLoc(),
6712 maskableOp->getResultTypes(), mask, passthru,
6713 maskableOp, createMaskOpRegion);
6714 return builder.create<MaskOp>(maskableOp->getLoc(),
6715 maskableOp->getResultTypes(), mask, maskableOp,