| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | merge_clusters.cc | 289 auto return_op = builder.create<mlir::tf_device::ReturnOp>( in CloneEmptyIfWithPredicate() local
543 auto return_op = in MergeClusters() local
|
| H A D | tpu_add_resource_device_attribute.cc | 70 auto return_op = llvm::dyn_cast<mlir::tf_device::ReturnOp>(use.getOwner()); in IdentifyConnectedAssignVariableOp() local
|
| H A D | layout_parsing.cc | 250 mlir::func::ReturnOp return_op, const int return_index) { in ExtractLayoutFromFunctionReturnAttr()
|
| H A D | mesh_propagation.cc | 284 if (auto return_op = llvm::dyn_cast<mlir::func::ReturnOp>(consumer)) { in InferMeshFromConsumers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | resource_op_lifting.cc | 615 auto return_op = func_op.front().getTerminator(); in FindResourceArgUseInfo() local
686 auto return_op = func_op.front().getTerminator(); in RemoveUnusedResourceArgumentsAndForwardedRetvals() local
831 auto return_op = body.front().getTerminator(); in HandleWhileLoop() local
|
| H A D | device_attribute_to_launch.cc | 48 auto* return_op = in WrapOpInLaunch() local
|
| H A D | materialize_mlir_passthrough_op.cc | 92 Operation &return_op = block.front(); in runOnOperation() local
|
| H A D | xla_inline_device_ops.cc | 45 auto& return_op = cluster_op.GetBody().front(); in InlineDeviceOp() local
|
| H A D | outside_compiled_to_host_launch.cc | 49 auto* return_op = in WrapOpInLaunch() local
|
| H A D | hoist_replicate_invariant_resource_writes.cc | 62 auto return_op = llvm::dyn_cast<tf_device::ReturnOp>( in MoveTailWritesAfterReplicate() local
|
| H A D | legalize_hlo.cc | 78 auto return_op = llvm::dyn_cast_or_null<mhlo::ReturnOp>(terminator); in ReplaceReturnOp() local
1547 auto return_op = dyn_cast_or_null<mhlo::ReturnOp>(&operations.back()); in MatchTopKComparator() local
1798 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in MatchBinaryReduceFunction() local
1820 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in MatchBinaryReduceFunction() local
2109 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in matchReduceComputation() local
|
| H A D | cluster_outlining.cc | 92 auto return_op = in BuildFunction() local
|
| H A D | promote_resources_to_args.cc | 175 auto return_op = in PromoteResourcesToArguments() local
|
| H A D | convert_control_to_data_outputs.cc | 231 func::ReturnOp return_op = cast<func::ReturnOp>(block.getTerminator()); in AppendFunctionResults() local
|
| H A D | tpu_cluster_formation.cc | 520 auto return_op = builder.create<tf_device::ReturnOp>(replicate_op.getLoc(), in ReplicateCluster() local
|
| H A D | shape_inference.cc | 2550 if (auto return_op = dyn_cast<func::ReturnOp>(block.getTerminator())) { in InferShapeForFunctionReturnType() local
2564 auto return_op = return_ops.front(); in InferShapeForFunctionReturnType() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | tf_functional_to_executor.cc | 69 auto return_op = dyn_cast<func::ReturnOp>(body.getTerminator()); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | reorder_assert.cc | 62 auto& return_op = block.back(); in ProcessFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | prepare_for_export.cc | 109 auto return_op = cast<mhlo::ReturnOp>(body_region.front().back()); in prepareWhileOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | functiondef_export.cc | 234 auto return_op = in ConvertGenericFunctionToFunctionDef() local
|
| H A D | graphdef_export.cc | 307 auto return_op = cast<ReturnOp>(func.getBody()->getTerminator()); in ExportFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | insert_main_function.cc | 120 auto& return_op = function.getBody().getBlocks().front().back(); in CreateMainFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| H A D | pass.cc | 304 Operation *return_op = func.getBody()->getTerminator(); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | pick_subgraphs.cc | 301 fn.walk([&](func::ReturnOp return_op) { in BuildSubgraphs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_device.cc | 699 Operation* return_op = body.getTerminator(); in EliminatePassThroughResults() local
|