Searched refs:control_ret_node_names (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.cc | 50 std::unordered_set<StringPiece, StringPieceHasher> control_ret_node_names; in FunctionBody() local
52 control_ret_node_names.insert(control_ret.second); in FunctionBody()
54 this->control_ret_nodes.reserve(control_ret_node_names.size()); in FunctionBody()
56 if (control_ret_node_names.count(n->name()) > 0) { in FunctionBody()
|
| D | function_optimization_registry.cc | 29 std::vector<std::string>* control_ret_node_names, in Run() argument
33 control_ret_node_names, in Run()
|
| D | function_optimization_registry.h | 48 std::vector<std::string>* control_ret_node_names,
65 std::vector<std::string>* control_ret_node_names,
|
| D | function_optimization_registry_pass_failure_test.cc | 33 std::vector<std::string>* control_ret_node_names, in Run() argument
|
| D | function_optimization_registry_test.cc | 34 std::vector<std::string>* control_ret_node_names, in Run() argument
|
| D | function_optimization_registration_test.cc | 32 std::vector<std::string>* control_ret_node_names, in Run() argument
|
| D | process_function_library_runtime.cc | 637 std::vector<string>* control_ret_node_names) { in GetGraphAndArgRets() argument
661 control_ret_node_names->reserve(fbody->control_ret_nodes.size()); in GetGraphAndArgRets()
663 control_ret_node_names->push_back(node->name()); in GetGraphAndArgRets()
715 std::vector<string> control_ret_node_names; in InstantiateMultiDevice() local
719 &ret_node_names, &ret_types, &control_ret_node_names)); in InstantiateMultiDevice()
770 &control_ret_node_names, &control_rets_updated)); in InstantiateMultiDevice()
776 for (const auto& control_ret : control_ret_node_names) { in InstantiateMultiDevice()
823 std::move(ret_node_names), std::move(control_ret_node_names), in InstantiateMultiDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 122 std::vector<std::string>* control_ret_node_names, in Run() argument
194 import_config.control_outputs = *control_ret_node_names; in Run()
309 control_ret_node_names->clear(); in Run()
310 control_ret_node_names->reserve(control_ret_nodes.size()); in Run()
312 control_ret_node_names->push_back(node->name()); in Run()
|
| D | mlir_graph_optimization_pass.h | 136 std::vector<std::string>* control_ret_node_names,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.h | 68 std::vector<string>* control_ret_node_names);
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_distributed_test.cc | 437 std::vector<std::string>* control_ret_node_names, in Run() argument
|