Searched refs:fetch_nodes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | graph_analyzer_tool.cc | 48 std::vector<string> fetch_nodes; in MaybePruneGraph() local
52 fetch_nodes.push_back(fetch); in MaybePruneGraph()
54 if (fetch_nodes.empty()) { in MaybePruneGraph()
59 fetch_nodes); in MaybePruneGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 192 const std::vector<Node*>& fetch_nodes, in PruneForTargets() argument
196 for (Node* n : fetch_nodes) { in PruneForTargets()
383 std::vector<Node*> fetch_nodes; in RewriteGraphForExecution() local
386 &fetch_nodes, &out_metadata->fetch_types)); in RewriteGraphForExecution()
391 if (!fetch_nodes.empty() || !target_node_names.empty()) { in RewriteGraphForExecution()
393 PruneForTargets(g, name_index, fetch_nodes, target_node_names)); in RewriteGraphForExecution()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 1308 std::vector<string> fetch_nodes = {"p2"}; in TEST_F() local
1310 item.graph, fetch_nodes, {{"v1", v1_t}, {"v2", v2_t}, {"v3", v3_t}}); in TEST_F()
1312 auto tensors = EvaluateNodes(output, fetch_nodes, in TEST_F()
1381 const std::vector<string> fetch_nodes = {"i1a", "i1b", "i2a", "i2b", in TEST_F() local
1384 item.graph, fetch_nodes, {{"v1", v1_t}, {"v2", v2_t}, {"v3", v3_t}}); in TEST_F()
1385 EXPECT_EQ(fetch_nodes.size(), tensors_expected.size()); in TEST_F()
1386 auto tensors = EvaluateNodes(output, fetch_nodes, in TEST_F()
1388 EXPECT_EQ(fetch_nodes.size(), tensors.size()); in TEST_F()
1389 for (int i = 0; i < fetch_nodes.size(); i++) in TEST_F()
1532 std::vector<string> fetch_nodes = {"m", "m2"}; in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.h | 53 std::set<string> fetch_nodes; member
|
| D | grappler_item_builder_test.cc | 327 config.fetch_nodes.insert("z"); in TEST_F()
|
| D | grappler_item_builder.cc | 205 for (const auto& fetch_node : cfg.fetch_nodes) { in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 308 const auto& fetch_nodes = grappler_item_->fetch; in Init() local
319 ComputeTransitiveFanin(graph, fetch_nodes, &ill_formed); in Init()
|