Home
last modified time | relevance | path

Searched refs:returnOp (Results 1 – 25 of 27) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DDropEquivalentBufferResults.cpp42 func::ReturnOp returnOp; in getAssumedUniqueReturnOp() local
45 if (returnOp) in getAssumedUniqueReturnOp()
47 returnOp = candidateOp; in getAssumedUniqueReturnOp()
50 return returnOp; in getAssumedUniqueReturnOp()
69 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in dropEquivalentBufferResults() local
71 if (!returnOp) in dropEquivalentBufferResults()
78 for (const auto &it : llvm::enumerate(returnOp.operands())) { in dropEquivalentBufferResults()
101 returnOp.operandsMutable().assign(newReturnValues); in dropEquivalentBufferResults()
H A DOneShotModuleBufferize.cpp110 func::ReturnOp returnOp; in getAssumedUniqueReturnOp() local
113 if (returnOp) in getAssumedUniqueReturnOp()
115 returnOp = candidateOp; in getAssumedUniqueReturnOp()
118 return returnOp; in getAssumedUniqueReturnOp()
151 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in aliasingFuncOpBBArgsAnalysis() local
152 assert(returnOp && "expected func with single return op"); in aliasingFuncOpBBArgsAnalysis()
154 for (OpOperand &returnVal : returnOp->getOpOperands()) in aliasingFuncOpBBArgsAnalysis()
289 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in getFuncOpsOrderedByCalls() local
290 if (!returnOp) in getFuncOpsOrderedByCalls()
339 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in foldMemRefCasts() local
[all …]
H A DFuncBufferizableOpInterfaceImpl.cpp48 func::ReturnOp returnOp; in getAssumedUniqueReturnOp() local
51 if (returnOp) in getAssumedUniqueReturnOp()
53 returnOp = candidateOp; in getAssumedUniqueReturnOp()
56 return returnOp; in getAssumedUniqueReturnOp()
373 auto returnOp = cast<func::ReturnOp>(op); in bufferize() local
374 assert(isa<FuncOp>(returnOp->getParentOp()) && in bufferize()
425 func::ReturnOp returnOp = getAssumedUniqueReturnOp(funcOp); in bufferize() local
426 assert(returnOp && "expected func with single return op"); in bufferize()
427 Location loc = returnOp.getLoc(); in bufferize()
464 rewriter.setInsertionPoint(returnOp); in bufferize()
[all …]
H A DOneShotAnalysis.cpp230 op->walk([&](Operation *returnOp) { in gatherYieldedTensors() argument
231 if (!isRegionReturnLike(returnOp) || !getOptions().isOpAllowed(returnOp)) in gatherYieldedTensors()
234 for (OpOperand &returnValOperand : returnOp->getOpOperands()) { in gatherYieldedTensors()
249 if (definingOp->getParentOp() == returnOp->getParentOp()) in gatherYieldedTensors()
900 op->walk([&](Operation *returnOp) { in assertDestinationPassingStyle() argument
901 if (!isRegionReturnLike(returnOp) || in assertDestinationPassingStyle()
902 !state.getOptions().isOpAllowed(returnOp)) in assertDestinationPassingStyle()
915 if (definingOp->isProperAncestor(returnOp)) in assertDestinationPassingStyle()
922 *returnOp->getParentOp())) in assertDestinationPassingStyle()
924 if (happensBefore(definingOp, returnOp, domInfo)) in assertDestinationPassingStyle()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/FuncToSPIRV/
H A DFuncToSPIRV.cpp45 matchAndRewrite(func::ReturnOp returnOp, OpAdaptor adaptor, in matchAndRewrite() argument
47 if (returnOp.getNumOperands() > 1) in matchAndRewrite()
50 if (returnOp.getNumOperands() == 1) { in matchAndRewrite()
52 returnOp, adaptor.getOperands()[0]); in matchAndRewrite()
54 rewriter.replaceOpWithNewOp<spirv::ReturnOp>(returnOp); in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Optimizer/Dialect/
H A DFIRDialect.cpp45 auto returnOp = llvm::cast<mlir::func::ReturnOp>(op); in handleTerminator() local
46 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
47 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/lib/Dialect/Func/IR/
H A DFuncOps.cpp75 auto returnOp = dyn_cast<ReturnOp>(op); in handleTerminator() local
76 if (!returnOp) in handleTerminator()
81 builder.create<cf::BranchOp>(op->getLoc(), newDest, returnOp.getOperands()); in handleTerminator()
90 auto returnOp = cast<ReturnOp>(op); in handleTerminator() local
93 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
94 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/
H A DMLIRGen.cpp156 ReturnOp returnOp; in mlirGen() local
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
159 if (!returnOp) { in mlirGen()
161 } else if (returnOp.hasOperand()) { in mlirGen()
/llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/
H A DMLIRGen.cpp156 ReturnOp returnOp; in mlirGen() local
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
159 if (!returnOp) { in mlirGen()
161 } else if (returnOp.hasOperand()) { in mlirGen()
/llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/
H A DMLIRGen.cpp156 ReturnOp returnOp; in mlirGen() local
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
159 if (!returnOp) { in mlirGen()
161 } else if (returnOp.hasOperand()) { in mlirGen()
H A DDialect.cpp67 auto returnOp = cast<ReturnOp>(op); in handleTerminator() local
70 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
71 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/
H A DMLIRGen.cpp156 ReturnOp returnOp; in mlirGen() local
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
159 if (!returnOp) { in mlirGen()
161 } else if (returnOp.hasOperand()) { in mlirGen()
H A DDialect.cpp67 auto returnOp = cast<ReturnOp>(op); in handleTerminator() local
70 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
71 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/
H A DMLIRGen.cpp156 ReturnOp returnOp; in mlirGen() local
158 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
159 if (!returnOp) { in mlirGen()
161 } else if (returnOp.hasOperand()) { in mlirGen()
H A DDialect.cpp67 auto returnOp = cast<ReturnOp>(op); in handleTerminator() local
70 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
71 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/
H A DMLIRGen.cpp211 ReturnOp returnOp; in mlirGen() local
213 returnOp = dyn_cast<ReturnOp>(entryBlock.back()); in mlirGen()
214 if (!returnOp) { in mlirGen()
216 } else if (returnOp.hasOperand()) { in mlirGen()
221 *returnOp.operand_type_begin())); in mlirGen()
H A DDialect.cpp68 auto returnOp = cast<ReturnOp>(op); in handleTerminator() local
71 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
72 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/
H A DGPUToSPIRV.cpp109 matchAndRewrite(gpu::ReturnOp returnOp, OpAdaptor adaptor,
335 gpu::ReturnOp returnOp, OpAdaptor adaptor, in matchAndRewrite() argument
340 rewriter.replaceOpWithNewOp<spirv::ReturnOp>(returnOp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp271 auto returnOp = dyn_cast<TestReturnOp>(op); in handleTerminator() local
272 if (!returnOp) in handleTerminator()
276 assert(returnOp.getNumOperands() == valuesToRepl.size()); in handleTerminator()
277 for (const auto &it : llvm::enumerate(returnOp.getOperands())) in handleTerminator()
921 Operation &returnOp = body.back().back(); in parse() local
922 result.types.append(returnOp.operand_type_begin(), in parse()
923 returnOp.operand_type_end()); in parse()
/llvm-project-15.0.7/mlir/lib/Target/Cpp/
H A DTranslateToCpp.cpp567 func::ReturnOp returnOp) { in printOperation() argument
570 switch (returnOp.getNumOperands()) { in printOperation()
574 os << " " << emitter.getOrCreateName(returnOp.getOperand(0)); in printOperation()
575 return success(emitter.hasValueInScope(returnOp.getOperand(0))); in printOperation()
578 if (failed(emitter.emitOperandsAndAttributes(*returnOp.getOperation()))) in printOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncToAsyncRuntime.cpp623 if (auto returnOp = dyn_cast<func::ReturnOp>(*terminator)) { in rewriteFuncAsCoroutine() local
624 ImplicitLocOpBuilder builder(loc, returnOp); in rewriteFuncAsCoroutine()
625 builder.create<YieldOp>(returnOp.getOperands()); in rewriteFuncAsCoroutine()
626 returnOp.erase(); in rewriteFuncAsCoroutine()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp213 funcOp.walk([&](func::ReturnOp returnOp) { in updateFunctionSignature() argument
214 for (const auto &operandEn : llvm::enumerate(returnOp.getOperands())) { in updateFunctionSignature()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp1324 if (auto returnOp = dyn_cast<tosa::YieldOp>(block.getTerminator())) in inferReturnTypeComponents() local
1325 yieldOps.push_back(returnOp); in inferReturnTypeComponents()
1367 if (auto returnOp = dyn_cast<tosa::YieldOp>(block.getTerminator())) in inferReturnTypeComponents() local
1368 yieldOps.push_back(returnOp); in inferReturnTypeComponents()
/llvm-project-15.0.7/mlir/docs/Tutorials/Toy/
H A DCh-4.md96 auto returnOp = cast<ReturnOp>(op);
99 assert(returnOp.getNumOperands() == valuesToRepl.size());
100 for (const auto &it : llvm::enumerate(returnOp.getOperands()))
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVDialect.cpp90 if (auto returnOp = dyn_cast<spirv::ReturnOp>(op)) { in handleTerminator() local

12