Home
last modified time | relevance | path

Searched refs:return_nodes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor_test.cc1734 opts.return_nodes.push_back("input"); in TEST_F()
1735 opts.return_nodes.push_back("t1"); in TEST_F()
1751 ASSERT_EQ(results.return_nodes.size(), 2); in TEST_F()
1754 EXPECT_EQ(results.return_nodes[0]->name(), "input"); in TEST_F()
1755 EXPECT_EQ(results.return_nodes[1]->name(), "t1"); in TEST_F()
1761 opts.return_nodes.push_back("input"); in TEST_F()
1766 ASSERT_EQ(results.return_nodes.size(), 1); in TEST_F()
1767 EXPECT_EQ(results.return_nodes[0]->name(), "import/input"); in TEST_F()
1773 opts.return_nodes.push_back("new_input"); in TEST_F()
1779 ASSERT_EQ(results.return_nodes.size(), 1); in TEST_F()
[all …]
Dgraph_constructor.cc113 return_nodes(in.return_nodes), in Options()
130 std::vector<string> return_nodes; member
162 std::vector<Node*>* return_nodes,
168 std::vector<Node*>* return_nodes,
174 std::vector<Node*>* return_nodes, in GraphConstructor() argument
182 return_nodes_(return_nodes), in GraphConstructor()
407 std::vector<Node*>* return_nodes, in NodeDefCopyingGraphConstructor() argument
409 : GraphConstructor(opts, g, refiner, return_tensors, return_nodes, in NodeDefCopyingGraphConstructor()
434 std::vector<Node*>* return_nodes, in NodeDefMovingGraphConstructor() argument
436 : GraphConstructor(opts, g, refiner, return_tensors, return_nodes, in NodeDefMovingGraphConstructor()
[all …]
Dgraph_constructor.h141 std::vector<string> return_nodes; member
168 std::vector<Node*> return_nodes; member
/external/tensorflow/tensorflow/c/
Dc_api.cc1674 opts->opts.return_nodes.push_back(oper_name); in TF_ImportGraphDefOptionsAddReturnOperation()
1679 return opts->opts.return_nodes.size(); in TF_ImportGraphDefOptionsNumReturnOperations()
1692 *num_opers = results->return_nodes.size(); in TF_ImportGraphDefResultsReturnOperations()
1693 *opers = results->return_nodes.data(); in TF_ImportGraphDefResultsReturnOperations()
1735 DCHECK(tf_results->return_nodes.empty()); in GraphImportGraphDefLocked()
1736 tf_results->return_nodes.resize(results.return_nodes.size()); in GraphImportGraphDefLocked()
1737 for (int i = 0; i < results.return_nodes.size(); ++i) { in GraphImportGraphDefLocked()
1738 tf_results->return_nodes[i] = ToOperation(results.return_nodes[i]); in GraphImportGraphDefLocked()
1847 std::vector<tensorflow::Output>* return_nodes) { in CopyGraph() argument
1848 DCHECK(return_nodes != nullptr); in CopyGraph()
[all …]
Dc_api_internal.h146 std::vector<TF_Operation*> return_nodes; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc417 auto& return_nodes = returns_[fetch]; in AddFetchNode() local
429 return_nodes.push_back(node); in AddFetchNode()