Home
last modified time | relevance | path

Searched refs:YieldOp (Results 1 – 25 of 93) sorted by relevance

1234

/llvm-project-15.0.7/mlir/test/python/dialects/
H A Dtransform_structured_ext.py24 transform.YieldOp()
35 transform.YieldOp()
50 transform.YieldOp()
63 transform.YieldOp()
80 transform.YieldOp()
96 transform.YieldOp()
108 transform.YieldOp()
119 transform.YieldOp()
135 transform.YieldOp()
148 transform.YieldOp()
[all …]
H A Dtransform.py22 transform.YieldOp([sequence.bodyTarget])
39 transform.YieldOp([doubly_nested.bodyTarget])
40 transform.YieldOp()
41 transform.YieldOp()
63 transform.YieldOp(match)
80 transform.YieldOp()
92 transform.YieldOp()
108 transform.YieldOp()
H A Dtransform_loop_ext.py24 transform.YieldOp()
35 transform.YieldOp()
46 transform.YieldOp()
56 transform.YieldOp()
68 transform.YieldOp()
H A Dscf.py29 scf.YieldOp(loop.inner_iter_args)
48 scf.YieldOp([loop.induction_variable])
70 scf.YieldOp(loop.inner_iter_args)
98 scf.YieldOp([])
120 scf.YieldOp([x_true, y_true])
124 scf.YieldOp([x_false, y_false])
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DStructuralTypeConversions.cpp123 class ConvertYieldOpTypes : public OpConversionPattern<scf::YieldOp> {
127 matchAndRewrite(scf::YieldOp op, OpAdaptor adaptor, in matchAndRewrite()
129 rewriter.replaceOpWithNewOp<scf::YieldOp>(op, adaptor.getOperands()); in matchAndRewrite()
186 target.addDynamicallyLegalOp<scf::YieldOp>([&](scf::YieldOp op) { in populateSCFStructuralTypeConversionsAndLegality()
H A DBufferizableOpInterfaceImpl.cpp58 auto yieldOp = dyn_cast<scf::YieldOp>( in getAliasingOpOperand()
166 auto thenYieldOp = cast<scf::YieldOp>(ifOp.thenBlock()->getTerminator()); in bufferize()
203 rewriter.replaceOpWithNewOp<scf::YieldOp>(thenYieldOp, thenResults); in bufferize()
205 rewriter.replaceOpWithNewOp<scf::YieldOp>(elseYieldOp, elseResults); in bufferize()
407 cast<scf::YieldOp>(forOp.getLoopBody().front().getTerminator()); in bufferRelation()
442 cast<scf::YieldOp>(forOp.getLoopBody().front().getTerminator()); in resolveConflicts()
539 cast<scf::YieldOp>(forOp.getLoopBody().front().getTerminator()); in verifyAnalysis()
854 scf::YieldOp> {
884 auto yieldOp = cast<scf::YieldOp>(op); in bufferize()
917 replaceOpWithNewBufferizedOp<scf::YieldOp>(rewriter, op, newResults); in bufferize()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/LinalgToLLVM/
H A DLinalgToLLVM.cpp56 class YieldOpConversion : public ConvertOpToLLVMPattern<linalg::YieldOp> {
58 using ConvertOpToLLVMPattern<linalg::YieldOp>::ConvertOpToLLVMPattern;
61 matchAndRewrite(linalg::YieldOp op, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToSCF/
H A DTosaToSCF.cpp33 auto yield = cast<YieldOp>(headBlock->getTerminator()); in inlineIfCase()
35 rewriter.create<scf::YieldOp>(yield.getLoc(), yield.getInputs()); in inlineIfCase()
49 auto yield = cast<YieldOp>(headBlock->getTerminator()); in inlineWhileCase()
58 rewriter.create<scf::YieldOp>(yield.getLoc(), yield.getInputs()); in inlineWhileCase()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp51 auto retValOp = dyn_cast<scf::YieldOp>(op); in handleTerminator()
76 builder.create<scf::YieldOp>(loc); in buildTerminatedBody()
218 if (YieldOp yieldOp = dyn_cast<YieldOp>(blk.getTerminator())) { in matchAndRewrite()
570 builder.create<scf::YieldOp>(loc, results); in buildLoopNest()
1893 YieldOp thenYield2 = cast<YieldOp>(nextThen->getTerminator()); in matchAndRewrite()
1915 YieldOp elseYield2 = cast<YieldOp>(nextElse->getTerminator()); in matchAndRewrite()
2068 rewriter.create<YieldOp>(loc, elseYield); in matchAndRewrite()
2086 YieldOp IfOp::thenYield() { return cast<YieldOp>(&thenBlock()->back()); } in thenYield()
2093 YieldOp IfOp::elseYield() { return cast<YieldOp>(&elseBlock()->back()); } in elseYield()
2564 YieldOp WhileOp::getYieldOp() { in getYieldOp()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.cpp92 b.create<scf::YieldOp>(loc, broadcastedDim); in getBroadcastedDim()
112 b.create<scf::YieldOp>(loc, dim); in getBroadcastedDim()
162 b.create<tensor::YieldOp>(loc, broadcastedDim); in matchAndRewrite()
294 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
313 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
318 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
424 b.create<scf::YieldOp>(loc, mappedResults); in matchAndRewrite()
513 b.create<scf::YieldOp>(loc, ValueRange({conjNext})); in matchAndRewrite()
515 b.create<scf::YieldOp>(loc, loop.getResults()); in matchAndRewrite()
520 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferSplitRewritePatterns.cpp262 b.create<scf::YieldOp>(loc, viewAndIndices); in createFullPartialLinalgCopy()
279 b.create<scf::YieldOp>(loc, viewAndIndices); in createFullPartialLinalgCopy()
317 b.create<scf::YieldOp>(loc, viewAndIndices); in createFullPartialVectorTransferRead()
332 b.create<scf::YieldOp>(loc, viewAndIndices); in createFullPartialVectorTransferRead()
369 b.create<scf::YieldOp>(loc, viewAndIndices); in getLocationToWriteFullVec()
377 b.create<scf::YieldOp>(loc, viewAndIndices); in getLocationToWriteFullVec()
407 b.create<scf::YieldOp>(loc, ValueRange{}); in createFullPartialLinalgCopy()
438 b.create<scf::YieldOp>(loc, ValueRange{}); in createFullPartialVectorTransferWrite()
H A DVectorDistribute.cpp81 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()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp418 b.create<scf::YieldOp>(loc); in createParallelComputeFunction()
547 executeBuilder.create<async::YieldOp>(executeLoc, ValueRange()); in createAsyncDispatchFunction()
554 b.create<scf::YieldOp>(ValueRange({start, midIndex})); in createAsyncDispatchFunction()
616 b.create<scf::YieldOp>(); in doAsyncDispatch()
639 b.create<scf::YieldOp>(); in doAsyncDispatch()
694 executeBuilder.create<async::YieldOp>(executeLoc, ValueRange()); in doSequentialDispatch()
701 b.create<scf::YieldOp>(); in doSequentialDispatch()
758 nestedBuilder.create<scf::YieldOp>(loc); in matchAndRewrite()
879 b.create<scf::YieldOp>(); in matchAndRewrite()
896 b.create<scf::YieldOp>(); in matchAndRewrite()
[all …]
H A DAsyncToAsyncRuntime.cpp188 if (auto yield = dyn_cast<YieldOp>(terminator)) { in setupCoroMachinery()
523 class YieldOpLowering : public OpConversionPattern<async::YieldOp> {
528 : OpConversionPattern<async::YieldOp>(ctx), in YieldOpLowering()
532 matchAndRewrite(async::YieldOp op, OpAdaptor adaptor, in matchAndRewrite()
625 builder.create<YieldOp>(returnOp.getOperands()); in rewriteFuncAsCoroutine()
788 runtimeTarget.addIllegalOp<ExecuteOp, AwaitOp, AwaitAllOp, async::YieldOp>(); in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToSPIRV/
H A DSCFToSPIRV.cpp102 class TerminatorOpConversion final : public SCFToSPIRVPattern<scf::YieldOp> {
104 using SCFToSPIRVPattern<scf::YieldOp>::SCFToSPIRVPattern;
107 matchAndRewrite(scf::YieldOp terminatorOp, OpAdaptor adaptor,
304 scf::YieldOp terminatorOp, OpAdaptor adaptor, in matchAndRewrite()
364 auto yield = cast<scf::YieldOp>(afterBlock.getTerminator()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DSplitPadding.cpp78 builder.create<scf::YieldOp>(loc, padOp.getSource()); in matchAndRewrite()
82 builder.create<scf::YieldOp>(loc, newOp->getResults()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/mlir-tblgen/
H A Dgen-dialect-doc.td16 def AOp : Op<Test_Dialect, "a", [NoSideEffect, SingleBlockImplicitTerminator<"YieldOp">]>;
33 // CHECK: Traits: SingleBlockImplicitTerminator<YieldOp>
/llvm-project-15.0.7/mlir/test/lib/Dialect/SCF/
H A DTestSCFUtils.cpp59 cast<scf::YieldOp>(block->getTerminator()).getResults()) { in runOnOperation()
160 rewriter.create<scf::YieldOp>(loc, op->getResults()); in predicateOp()
168 rewriter.create<scf::YieldOp>(loc, zeros); in predicateOp()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp427 if (auto terminator = dyn_cast<scf::YieldOp>(block.getTerminator())) { in matchAndRewrite()
480 rewriter.create<scf::YieldOp>(loc, forOp.getResults()); in matchAndRewrite()
515 rewriter.create<scf::YieldOp>(loc, yieldOperands); in matchAndRewrite()
555 auto yieldOp = cast<scf::YieldOp>(afterLast->getTerminator()); in matchAndRewrite()
580 auto yield = dyn_cast<scf::YieldOp>(&afterBlock.front()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp54 auto yieldOp = cast<linalg::YieldOp>(op.region().front().getTerminator()); in isSampling()
81 auto yieldOp = cast<linalg::YieldOp>(op.region().front().getTerminator()); in isSumOfMul()
182 rewriter.create<linalg::YieldOp>(loc, last); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToOpenMP/
H A DSCFToOpenMP.cpp195 builder.create<omp::YieldOp>(reduce.getLoc(), init); in createDecl()
201 builder.replaceOpWithNewOp<omp::YieldOp>(terminator, in createDecl()
228 builder.create<omp::YieldOp>(reduce.getLoc(), ArrayRef<Value>()); in addAtomicRMW()
404 rewriter.create<omp::YieldOp>(loc, ValueRange()); in matchAndRewrite()
407 auto oldYield = cast<scf::YieldOp>(scopeBlock->getTerminator()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp239 builder.create<tensor::YieldOp>(gLoc, padValue); in bubbleUpPadSlice()
273 b.create<scf::YieldOp>(loc, createGenerateOp()->getResult(0)); in bubbleUpPadSlice()
277 b.create<scf::YieldOp>(loc, createPadOfExtractSlice()->getResult(0)); in bubbleUpPadSlice()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/IR/
H A DAsync.cpp36 LogicalResult YieldOp::verify() { in verify()
52 YieldOp::getMutableSuccessorOperands(Optional<unsigned> index) { in getMutableSuccessorOperands()
128 builder.create<async::YieldOp>(result.location, ValueRange()); in build()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp283 YieldOp yield = dyn_cast<YieldOp>(term); in verifyNumBlockArgs()
376 LogicalResult YieldOp::verify() { in verify()
/llvm-project-15.0.7/mlir/lib/Dialect/OpenMP/IR/
H A DOpenMPDialect.cpp613 for (YieldOp yieldOp : initializerRegion().getOps<YieldOp>()) { in verifyRegions()
629 for (YieldOp yieldOp : reductionRegion().getOps<YieldOp>()) { in verifyRegions()
865 YieldOp yieldOp = *region().getOps<YieldOp>().begin(); in verifyRegions()

1234