Searched defs:return_op (Results 1 – 13 of 13) sorted by relevance
53 auto return_op = dyn_cast<mhlo::ReturnOp>(block->getTerminator()); in ReplaceTerminators() local165 auto return_op = dyn_cast<mhlo::ReturnOp>(new_block->getTerminator()); in LowerWhileOp() local194 auto return_op = dyn_cast<mlir::mhlo::ReturnOp>(new_block->getTerminator()); in LowerWhileOp() local
442 auto return_op = cast<mhlo::ReturnOp>(entry_block.getTerminator()); in matchAndRewrite() local
65 auto return_op = dyn_cast<ReturnOp>(body.getTerminator()); in runOnFunction() local
87 Operation &return_op = block.front(); in runOnFunction() local
71 auto* return_op = in WrapOpInLaunch() local
194 auto return_op = llvm::dyn_cast_or_null<ReturnOp>(block.getTerminator()); in PromoteResourcesToArguments() local
683 auto return_op = func_op.front().getTerminator(); in FindResourceArgUseInfo() local754 auto return_op = func_op.front().getTerminator(); in RemoveUnusedResourceArgumentsAndForwardedRetvals() local898 auto return_op = body.front().getTerminator(); in HandleWhileLoop() local
575 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in MatchBinaryReduceFunction() local597 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in MatchBinaryReduceFunction() local
485 auto return_op = builder.create<tf_device::ReturnOp>(replicate_op.getLoc(), in ReplicateCluster() local
1463 if (auto return_op = dyn_cast<ReturnOp>(block.getTerminator())) { in InferShapeForFunctionReturnType() local1474 auto return_op = return_ops.front(); in InferShapeForFunctionReturnType() local
599 Operation &return_op = body.back().back(); in parseWrappingRegionOp() local
1522 auto return_op = builder_.create<ReturnOp>(builder_.getUnknownLoc()); in Initialize() local
2119 WalkResult walker = branch_region.walk([&](ReturnOp return_op) { in Verify()