Home
last modified time | relevance | path

Searched refs:control_ret_node_names (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc483 std::vector<string>* control_ret_node_names) { in GetGraphAndRets() argument
503 control_ret_node_names->reserve(fbody->control_ret_nodes.size()); in GetGraphAndRets()
505 control_ret_node_names->push_back(node->name()); in GetGraphAndRets()
555 std::vector<string> control_ret_node_names; in InstantiateMultiDevice() local
559 &control_ret_node_names)); in InstantiateMultiDevice()
628 std::move(ret_node_names), std::move(control_ret_node_names), in InstantiateMultiDevice()
Dfunction.cc1944 std::unordered_set<StringPiece, StringPieceHasher> control_ret_node_names; in FunctionBody() local
1946 control_ret_node_names.insert(control_ret.second); in FunctionBody()
1948 this->control_ret_nodes.reserve(control_ret_node_names.size()); in FunctionBody()
1950 if (control_ret_node_names.count(n->name()) > 0) { in FunctionBody()