Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor_test.cc1730 opts.return_nodes.push_back("input"); in TEST_F()
1731 opts.return_nodes.push_back("t1"); in TEST_F()
1747 ASSERT_EQ(results.return_nodes.size(), 2); in TEST_F()
1750 EXPECT_EQ(results.return_nodes[0]->name(), "input"); in TEST_F()
1751 EXPECT_EQ(results.return_nodes[1]->name(), "t1"); in TEST_F()
1757 opts.return_nodes.push_back("input"); in TEST_F()
1762 ASSERT_EQ(results.return_nodes.size(), 1); in TEST_F()
1763 EXPECT_EQ(results.return_nodes[0]->name(), "import/input"); in TEST_F()
1769 opts.return_nodes.push_back("new_input"); in TEST_F()
1775 ASSERT_EQ(results.return_nodes.size(), 1); in TEST_F()
[all …]
Dgraph_constructor.h129 std::vector<string> return_nodes; member
156 std::vector<Node*> return_nodes; member
Dgraph_constructor.cc88 return_nodes(in.return_nodes), in Options()
104 std::vector<string> return_nodes; member
128 std::vector<Node*>* return_nodes, in Construct() argument
136 return_tensors, return_nodes, in Construct()
149 std::vector<Node*>* return_nodes, in GraphConstructor() argument
160 return_nodes_(return_nodes), in GraphConstructor()
1141 if (opts_.return_nodes.empty()) return Status::OK(); in PopulateReturnNodes()
1142 for (StringPiece name : opts_.return_nodes) { in PopulateReturnNodes()
1241 if (!opts.return_nodes.empty()) { in ImportGraphDef()
1255 if (!results->return_tensors.empty() || !results->return_nodes.empty() || in ImportGraphDef()
[all …]
/external/tensorflow/tensorflow/c/
Dc_api.cc2081 opts->opts.return_nodes.push_back(oper_name); in TF_ImportGraphDefOptionsAddReturnOperation()
2086 return opts->opts.return_nodes.size(); in TF_ImportGraphDefOptionsNumReturnOperations()
2099 *num_opers = results->return_nodes.size(); in TF_ImportGraphDefResultsReturnOperations()
2100 *opers = results->return_nodes.data(); in TF_ImportGraphDefResultsReturnOperations()
2142 DCHECK(tf_results->return_nodes.empty()); in GraphImportGraphDefLocked()
2143 tf_results->return_nodes.resize(results.return_nodes.size()); in GraphImportGraphDefLocked()
2144 for (int i = 0; i < results.return_nodes.size(); ++i) { in GraphImportGraphDefLocked()
2145 tf_results->return_nodes[i] = ToOperation(results.return_nodes[i]); in GraphImportGraphDefLocked()
2254 std::vector<tensorflow::Output>* return_nodes) { in CopyGraph() argument
2255 DCHECK(return_nodes != nullptr); in CopyGraph()
[all …]
Dc_api_internal.h154 std::vector<TF_Operation*> return_nodes; member
/external/v8/src/wasm/
Dfunction-body-decoder.cc788 TFNode** return_nodes = nullptr; in DoCall() local
794 builder_->CallIndirect(index, arg_nodes, &return_nodes, in DoCall()
797 builder_->CallDirect(index, arg_nodes, &return_nodes, in DoCall()
802 returns[i].node = return_nodes[i]; in DoCall()