Lines Matching refs:YieldOp

81   auto yieldOp = cast<vector::YieldOp>(ifOp.thenBlock()->getTerminator());  in rewriteWarpOpToScfFor()
132   rewriter.create<scf::YieldOp>(yieldLoc);  in rewriteWarpOpToScfFor()
160       cast<vector::YieldOp>(newOpBody.getBlocks().begin()->getTerminator());  in moveRegionToNewWarpOpAndReplaceReturns()
175   auto yield = cast<vector::YieldOp>(  in moveRegionToNewWarpOpAndAppendReturns()
212   auto yield = cast<vector::YieldOp>(  in getWarpResult()
400           return isa<vector::TransferWriteOp, vector::YieldOp>(&op);  in tryExtractOp()
420     rewriter.create<vector::YieldOp>(newWarpOp.getLoc());  in tryExtractOp()
632     auto yield = cast<vector::YieldOp>(  in matchAndRewrite()
663     auto yield = cast<vector::YieldOp>(  in matchAndRewrite()
785     auto yield = cast<vector::YieldOp>(  in matchAndRewrite()
827     rewriter.create<vector::YieldOp>(innerWarp.getLoc(), yieldOperands);  in matchAndRewrite()
830       rewriter.create<scf::YieldOp>(forOp.getLoc(), innerWarp.getResults());  in matchAndRewrite()