| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | functions.cc | 44 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem()
102 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs() function in tensorflow::grappler::GrapplerFunctionItem
282 std::vector<ControlOutput> control_outputs; in MakeGrapplerFunctionItem() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | graph_to_functiondef.cc | 318 std::vector<const Node*> control_outputs; in GraphToFunctionDefHelper() local
375 const std::vector<const Node*>& control_outputs, in GraphToFunctionDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition.cc | 220 const std::vector<std::string>& control_outputs, Graph* subgraph, in BuildPartitionedCallOp()
423 const std::vector<std::string>& control_outputs, in InsertTransferOps()
|
| H A D | tfrt_graph_execution_state.cc | 299 const std::vector<std::string>& control_outputs, in AdjustDeviceAssignment()
342 const std::vector<std::string>& control_outputs, in BuildXlaOpsAndMaybeInsertTransferOps()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | mlir_roundtrip_flags.h | 61 std::vector<string> control_outputs; member
|
| H A D | import_model.cc | 2459 auto control_outputs = in Convert() local
2494 auto control_outputs = in Convert() local
2724 llvm::ArrayRef<std::string> control_outputs, in GetControlRetsFromGraph()
3893 const std::vector<std::string> control_outputs, in ConvertGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 559 std::vector<TF_Operation*> control_outputs( in TF_OperationGetControlOutputs_wrapper() local
584 const std::vector<TF_Operation*>* control_outputs, in TF_GraphToFunction_wrapper()
|
| H A D | tf_session_wrapper.cc | 205 const char* description) { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | functiondef_export.cc | 196 if (auto control_outputs = in ConvertGenericFunctionToFunctionDef() local
|
| H A D | functiondef_import.cc | 280 SmallVector<Attribute> control_outputs; in ImportGenericFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_xla_computations_pass.cc | 285 absl::flat_hash_set<Node*> control_outputs; in BuildXlaLaunchOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function.cc | 156 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs()
|
| H A D | c_api.cc | 1208 TF_Operation** control_outputs, in TF_OperationGetControlOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | dependency_optimizer.cc | 542 std::vector<gtl::InlinedVector<std::pair<int, int>, 2>> control_outputs( in TransitiveReduction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 922 const std::vector<Node*>& control_outputs, in CreateXlaSplitOp()
978 const std::vector<Node*>& control_outputs, const DataType dtype, in ShardInputWithXlaSplitOp()
1078 std::vector<Node*> control_outputs; in CreateOrGetXlaSplitNodeForVariableArg() local
|
| H A D | encapsulate_tpu_computations_pass.cc | 2724 gtl::FlatSet<Node*> control_outputs; in BuildTPUReplicateOps() local
|