| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| graph_analyzer_tool.cc | 48 std::vector<string> fetch_nodes; local
52 fetch_nodes.push_back(fetch);
54 if (fetch_nodes.empty()) {
59 fetch_nodes);
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.h | 53 std::set<string> fetch_nodes; member in struct:tensorflow::grappler::ItemConfig
|
| grappler_item_builder_test.cc | 327 config.fetch_nodes.insert("z");
|
| grappler_item_builder.cc | 205 for (const auto& fetch_node : cfg.fetch_nodes) {
|
| /external/tensorflow/tensorflow/core/graph/ |
| subgraph.cc | 192 const std::vector<Node*>& fetch_nodes,
196 for (Node* n : fetch_nodes) {
383 std::vector<Node*> fetch_nodes; local
386 &fetch_nodes, &out_metadata->fetch_types));
391 if (!fetch_nodes.empty() || !target_node_names.empty()) {
393 PruneForTargets(g, name_index, fetch_nodes, target_node_names));
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding_test.cc | 1308 std::vector<string> fetch_nodes = {"p2"}; local
1381 const std::vector<string> fetch_nodes = {"i1a", "i1b", "i2a", "i2b", local
1532 std::vector<string> fetch_nodes = {"m", "m2"}; local
2672 const std::vector<string> fetch_nodes = {"i1", "i2"}; local
2715 std::vector<string> fetch_nodes = {"o1", "o2", "p1", "p2"}; local
2786 std::vector<string> fetch_nodes = {"o1", "o2"}; local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_scheduler.cc | 308 const auto& fetch_nodes = grappler_item_->fetch; local
316 // Get the nodes that would run to output fetch_nodes.
319 ComputeTransitiveFanin(graph, fetch_nodes, &ill_formed);
[all...] |
