Searched refs:yieldOp (Results 1 – 9 of 9) sorted by relevance
147 auto &yieldOp = executeOp.getBody()->getOperations().back(); in ~DeferWaitCallback() local148 yieldOp.insertOperands(yieldOp.getNumOperands(), in ~DeferWaitCallback()
80 auto yieldOp = cast<YieldOp>(op.getBody()->getTerminator()); in matchAndRewrite() local81 b.create<StoreOp>(loc, mapping.lookup(yieldOp.value()), result, ivs); in matchAndRewrite()
512 auto yieldOp = cast<scf::YieldOp>(afterLast->getTerminator()); in matchAndRewrite() local513 rewriter.replaceOpWithNewOp<BranchOp>(yieldOp, before, yieldOp.results()); in matchAndRewrite()
423 auto yieldOp = cast<scf::YieldOp>(block.getTerminator()); in matchAndRewrite() local431 keepMask.reserve(yieldOp.getNumOperands()); in matchAndRewrite()437 newYieldValues.reserve(yieldOp.getNumOperands()); in matchAndRewrite()441 yieldOp.getOperands() // iter yield in matchAndRewrite()729 auto yieldOp = cast<scf::YieldOp>(dest->getTerminator()); in transferBody() local733 return yieldOp.getOperand(result.getResultNumber()); in transferBody()735 rewriter.updateRootInPlace(yieldOp, in transferBody()736 [&]() { yieldOp->setOperands(usedOperands); }); in transferBody()
168 auto yieldOp = dyn_cast<linalg::YieldOp>(scalarOp); in vectorize() local169 if (yieldOp) { in vectorize()171 llvm::zip(yieldOp.values(), generic.getOutputBuffers())) { in vectorize()
164 if (auto yieldOp = dyn_cast<linalg::YieldOp>(op)) { in generateFusedTensorOpRegion() local166 Value yieldVal = yieldOp.getOperand(0); in generateFusedTensorOpRegion()
260 auto &yieldOp = assumingBlock->back(); in inlineRegionIntoParent() local262 rewriter.replaceOp(op, yieldOp.getOperands()); in inlineRegionIntoParent()263 rewriter.eraseOp(&yieldOp); in inlineRegionIntoParent()
3402 static LogicalResult verify(spirv::YieldOp yieldOp) { in verify() argument3403 Operation *parentOp = yieldOp->getParentOp(); in verify()3406 return yieldOp.emitOpError( in verify()3412 if (yieldOp.getOperand().getDefiningOp() != &enclosedOp) in verify()3413 return yieldOp.emitOpError( in verify()3436 Operation &yieldOp = block.getOperations().back(); in verify() local3438 if (!isa<spirv::YieldOp>(yieldOp)) in verify()
1773 auto yieldOp = in verify() local1775 if (yieldOp.value().getType() != resultTy.getElementType()) in verify()