| /llvm-project-15.0.7/mlir/lib/Conversion/FuncToSPIRV/ |
| H A D | FuncToSPIRV.cpp | 40 class ReturnOpPattern final : public OpConversionPattern<func::ReturnOp> {
42 using OpConversionPattern<func::ReturnOp>::OpConversionPattern;
45 matchAndRewrite(func::ReturnOp returnOp, OpAdaptor adaptor, in matchAndRewrite()
54 rewriter.replaceOpWithNewOp<spirv::ReturnOp>(returnOp); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUOpsLowering.h | 70 struct GPUReturnOpLowering : public ConvertOpToLLVMPattern<gpu::ReturnOp> {
71 using ConvertOpToLLVMPattern<gpu::ReturnOp>::ConvertOpToLLVMPattern;
74 matchAndRewrite(gpu::ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
76 rewriter.replaceOpWithNewOp<LLVM::ReturnOp>(op, adaptor.getOperands()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | DropEquivalentBufferResults.cpp | 41 static func::ReturnOp getAssumedUniqueReturnOp(func::FuncOp funcOp) { in getAssumedUniqueReturnOp()
42 func::ReturnOp returnOp; in getAssumedUniqueReturnOp()
44 if (auto candidateOp = dyn_cast<func::ReturnOp>(b.getTerminator())) { in getAssumedUniqueReturnOp()
69 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in dropEquivalentBufferResults()
|
| H A D | OneShotModuleBufferize.cpp | 109 static func::ReturnOp getAssumedUniqueReturnOp(func::FuncOp funcOp) { in getAssumedUniqueReturnOp()
110 func::ReturnOp returnOp; in getAssumedUniqueReturnOp()
112 if (auto candidateOp = dyn_cast<func::ReturnOp>(b.getTerminator())) { in getAssumedUniqueReturnOp()
151 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in aliasingFuncOpBBArgsAnalysis()
289 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in getFuncOpsOrderedByCalls()
339 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in foldMemRefCasts()
|
| H A D | FuncBufferizableOpInterfaceImpl.cpp | 47 static func::ReturnOp getAssumedUniqueReturnOp(FuncOp funcOp) { in getAssumedUniqueReturnOp()
48 func::ReturnOp returnOp; in getAssumedUniqueReturnOp()
50 if (auto candidateOp = dyn_cast<func::ReturnOp>(b.getTerminator())) { in getAssumedUniqueReturnOp()
354 func::ReturnOp> {
373 auto returnOp = cast<func::ReturnOp>(op); in bufferize()
425 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in bufferize()
522 func::ReturnOp::attachInterface<func_ext::ReturnOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
|
| /llvm-project-15.0.7/mlir/test/python/dialects/ |
| H A D | builtin.py | 95 func.ReturnOp([b])
149 func.ReturnOp([f.entry_block.arguments[0]])
162 body_builder=lambda f: func.ReturnOp(
186 func.ReturnOp(f.arguments)
201 func.ReturnOp([])
|
| H A D | ml_program.py | 28 ml_program.ReturnOp([block.arguments[0]])
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Func/Transforms/ |
| H A D | FuncConversions.cpp | 96 class ReturnOpTypeConversion : public OpConversionPattern<ReturnOp> {
98 using OpConversionPattern<ReturnOp>::OpConversionPattern;
101 matchAndRewrite(ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
146 if (dyn_cast<ReturnOp>(op) && !returnOpAlwaysLegal) in isLegalForReturnOpTypeConversionPattern()
|
| H A D | DecomposeCallGraphTypes.cpp | 106 : public DecomposeCallGraphTypesOpConversionPattern<ReturnOp> {
110 matchAndRewrite(ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
116 rewriter.replaceOpWithNewOp<ReturnOp>(op, newOperands); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/FuncToLLVM/ |
| H A D | FuncToLLVM.cpp | 171 rewriter.create<LLVM::ReturnOp>(loc, ValueRange{}); in wrapForExternalCallers()
173 rewriter.create<LLVM::ReturnOp>(loc, call.getResults()); in wrapForExternalCallers()
272 builder.create<LLVM::ReturnOp>(loc, ValueRange{result}); in wrapExternalFunction()
274 builder.create<LLVM::ReturnOp>(loc, call.getResults()); in wrapExternalFunction()
592 struct ReturnOpLowering : public ConvertOpToLLVMPattern<func::ReturnOp> {
593 using ConvertOpToLLVMPattern<func::ReturnOp>::ConvertOpToLLVMPattern;
596 matchAndRewrite(func::ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
628 rewriter.replaceOpWithNewOp<LLVM::ReturnOp>(op, TypeRange(), ValueRange(), in matchAndRewrite()
633 rewriter.replaceOpWithNewOp<LLVM::ReturnOp>( in matchAndRewrite()
649 rewriter.replaceOpWithNewOp<LLVM::ReturnOp>(op, TypeRange(), packed, in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Transforms/ |
| H A D | AbstractResult.cpp | 136 class ReturnOpConversion : public mlir::OpRewritePattern<mlir::func::ReturnOp> {
142 matchAndRewrite(mlir::func::ReturnOp ret, in matchAndRewrite()
162 rewriter.replaceOpWithNewOp<mlir::func::ReturnOp>(ret); in matchAndRewrite()
221 target.addDynamicallyLegalOp<mlir::func::ReturnOp>( in runOnOperation()
222 [](mlir::func::ReturnOp ret) { return ret.operands().empty(); }); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/ |
| H A D | GPUToSPIRV.cpp | 104 class GPUReturnOpConversion final : public OpConversionPattern<gpu::ReturnOp> {
106 using OpConversionPattern<gpu::ReturnOp>::OpConversionPattern;
109 matchAndRewrite(gpu::ReturnOp returnOp, OpAdaptor adaptor,
335 gpu::ReturnOp returnOp, OpAdaptor adaptor, in matchAndRewrite()
340 rewriter.replaceOpWithNewOp<spirv::ReturnOp>(returnOp); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/ |
| H A D | LowerToAffineLoops.cpp | 255 struct ReturnOpLowering : public OpRewritePattern<toy::ReturnOp> {
256 using OpRewritePattern<toy::ReturnOp>::OpRewritePattern;
258 LogicalResult matchAndRewrite(toy::ReturnOp op, in matchAndRewrite()
266 rewriter.replaceOpWithNewOp<func::ReturnOp>(op); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/ |
| H A D | LowerToAffineLoops.cpp | 255 struct ReturnOpLowering : public OpRewritePattern<toy::ReturnOp> {
256 using OpRewritePattern<toy::ReturnOp>::OpRewritePattern;
258 LogicalResult matchAndRewrite(toy::ReturnOp op, in matchAndRewrite()
266 rewriter.replaceOpWithNewOp<func::ReturnOp>(op); in matchAndRewrite()
|
| H A D | MLIRGen.cpp | 156 ReturnOp returnOp; in mlirGen()
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
160 builder.create<ReturnOp>(loc(funcAST.getProto()->loc())); in mlirGen()
233 builder.create<ReturnOp>(location, expr ? makeArrayRef(expr) in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/ |
| H A D | LowerToAffineLoops.cpp | 255 struct ReturnOpLowering : public OpRewritePattern<toy::ReturnOp> {
256 using OpRewritePattern<toy::ReturnOp>::OpRewritePattern;
258 LogicalResult matchAndRewrite(toy::ReturnOp op, in matchAndRewrite()
266 rewriter.replaceOpWithNewOp<func::ReturnOp>(op); in matchAndRewrite()
|
| H A D | MLIRGen.cpp | 156 ReturnOp returnOp; in mlirGen()
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
160 builder.create<ReturnOp>(loc(funcAST.getProto()->loc())); in mlirGen()
233 builder.create<ReturnOp>(location, expr ? makeArrayRef(expr) in mlirGen()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Func/ |
| H A D | TestDecomposeCallGraphTypes.cpp | 47 target.addDynamicallyLegalOp<func::ReturnOp>([&](func::ReturnOp op) { in runOnOperation()
|
| /llvm-project-15.0.7/mlir/test/python/ir/ |
| H A D | blocks.py | 41 func.ReturnOp([])
68 func.ReturnOp([])
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/ |
| H A D | MLIRGen.cpp | 156 ReturnOp returnOp; in mlirGen()
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
160 builder.create<ReturnOp>(loc(funcAST.getProto()->loc())); in mlirGen()
229 builder.create<ReturnOp>(location, expr ? makeArrayRef(expr) in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/ |
| H A D | MLIRGen.cpp | 156 ReturnOp returnOp; in mlirGen()
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
160 builder.create<ReturnOp>(loc(funcAST.getProto()->loc())); in mlirGen()
229 builder.create<ReturnOp>(location, expr ? makeArrayRef(expr) in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/ |
| H A D | MLIRGen.cpp | 156 ReturnOp returnOp; in mlirGen()
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
160 builder.create<ReturnOp>(loc(funcAST.getProto()->loc())); in mlirGen()
233 builder.create<ReturnOp>(location, expr ? makeArrayRef(expr) in mlirGen()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaInferShapes.cpp | 226 if (isa<func::ReturnOp>(user)) in propagateShapesInRegion()
289 func.walk([&](func::ReturnOp op) { in runOnOperation()
317 rewriter.replaceOpWithNewOp<func::ReturnOp>(op, castedValues); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Func/IR/ |
| H A D | FuncOps.cpp | 75 auto returnOp = dyn_cast<ReturnOp>(op); in handleTerminator()
90 auto returnOp = cast<ReturnOp>(op); in handleTerminator()
346 LogicalResult ReturnOp::verify() { in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorPasses.cpp | 108 target.addDynamicallyLegalOp<func::ReturnOp>([&](func::ReturnOp op) { in runOnOperation()
|