Home
last modified time | relevance | path

Searched refs:return_tensors (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor_test.cc1636 opts.return_tensors.push_back({"input", 1}); in TEST_F()
1637 opts.return_tensors.push_back({"t1", 0}); in TEST_F()
1638 opts.return_tensors.push_back({"input", 0}); in TEST_F()
1652 ASSERT_EQ(results.return_tensors.size(), 3); in TEST_F()
1653 EXPECT_EQ(results.return_tensors[0].first->name(), "input"); in TEST_F()
1654 EXPECT_EQ(results.return_tensors[0].second, 1); in TEST_F()
1655 EXPECT_EQ(results.return_tensors[1].first->name(), "t1"); in TEST_F()
1656 EXPECT_EQ(results.return_tensors[1].second, 0); in TEST_F()
1657 EXPECT_EQ(results.return_tensors[2].first->name(), "input"); in TEST_F()
1658 EXPECT_EQ(results.return_tensors[2].second, 0); in TEST_F()
[all …]
Dgraph_constructor.h115 std::vector<SafeTensorId> return_tensors; member
153 std::vector<std::pair<Node*, Index>> return_tensors; member
Dgraph_constructor.cc87 return_tensors(in.return_tensors.begin(), in.return_tensors.end()), in Options()
103 std::vector<TensorId> return_tensors; member
127 std::vector<std::pair<Node*, int>>* return_tensors, in Construct() argument
136 return_tensors, return_nodes, in Construct()
148 std::vector<std::pair<Node*, int>>* return_tensors, in GraphConstructor() argument
159 return_tensors_(return_tensors), in GraphConstructor()
1110 if (opts_.return_tensors.empty()) return Status::OK(); in PopulateReturnTensors()
1111 for (const TensorId& id : opts_.return_tensors) { in PopulateReturnTensors()
1233 if (!opts.return_tensors.empty()) { in ImportGraphDef()
1255 if (!results->return_tensors.empty() || !results->return_nodes.empty() || in ImportGraphDef()
[all …]
/external/tensorflow/tensorflow/c/
Dc_api.cc2071 opts->opts.return_tensors.emplace_back(oper_name_str, index); in TF_ImportGraphDefOptionsAddReturnOutput()
2076 return opts->opts.return_tensors.size(); in TF_ImportGraphDefOptionsNumReturnOutputs()
2092 *num_outputs = results->return_tensors.size(); in TF_ImportGraphDefResultsReturnOutputs()
2093 *outputs = results->return_tensors.data(); in TF_ImportGraphDefResultsReturnOutputs()
2133 DCHECK(tf_results->return_tensors.empty()); in GraphImportGraphDefLocked()
2134 tf_results->return_tensors.resize(results.return_tensors.size()); in GraphImportGraphDefLocked()
2135 for (int i = 0; i < results.return_tensors.size(); ++i) { in GraphImportGraphDefLocked()
2136 tf_results->return_tensors[i].oper = in GraphImportGraphDefLocked()
2137 ToOperation(results.return_tensors[i].first); in GraphImportGraphDefLocked()
2138 tf_results->return_tensors[i].index = results.return_tensors[i].second; in GraphImportGraphDefLocked()
[all …]
Dc_api_internal.h153 std::vector<TF_Output> return_tensors; member