| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item.cc | 122 std::unordered_map<string, const NodeDef*> name_to_node; local
125 name_to_node[node.name()] = &node;
134 const NodeDef* node = name_to_node[NodeName(root)];
155 const NodeDef* in = name_to_node[NodeName(input)];
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| scc_test.cc | 86 std::unordered_map<string, const NodeDef*> name_to_node; local
89 name_to_node[n.name()] = &n;
100 EXPECT_EQ(components[name_to_node[pair.first]],
101 components[name_to_node[pair.second]]);
105 EXPECT_EQ(-1, components[name_to_node[node]]);
149 std::unordered_map<string, const NodeDef*> name_to_node; local
152 name_to_node[n.name()] = &n;
163 EXPECT_EQ(components[name_to_node[pair.first]],
164 components[name_to_node[pair.second]]);
171 EXPECT_NE(components[name_to_node[pair.first]]
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla_util.cc | 120 std::unordered_map<string, const NodeDef*> name_to_node; local
122 name_to_node[graph_def->node(i).name()] = &graph_def->node(i);
129 auto node_it = name_to_node.find(feed_id.node_name());
130 if (node_it == name_to_node.end()) {
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils_test.cc | 70 std::unordered_map<string, const NodeDef*> name_to_node; local
73 name_to_node["input"] = input;
76 name_to_node["filter"] = filter;
80 name_to_node["output_backprop"] = output_backprop;
84 name_to_node["input_sizes"] = input;
89 name_to_node["filter_sizes"] = filter_sizes;
138 OpInfo info = BuildOpInfoWithoutDevice(node, name_to_node, inputs);
158 std::unordered_map<string, const NodeDef*> name_to_node; local
160 name_to_node[node.name()] = &node;
167 OpInfo info = BuildOpInfoWithoutDevice(node, name_to_node, inputs)
[all...] |
| utils.cc | 84 const std::unordered_map<string, const NodeDef*>& name_to_node,
102 auto iter = name_to_node.find(input_node_name);
103 if (iter == name_to_node.end()) continue;
168 const std::unordered_map<string, const NodeDef*>& name_to_node) {
222 const std::unordered_map<string, const NodeDef*>& name_to_node,
230 ExtractExtraProperties(node, name_to_node, &op_info);
249 std::unordered_map<string, const NodeDef*> name_to_node; local
254 name_to_node[node.name()] = &node;
273 FindInputFeatures(node, name_to_cost, name_to_node);
274 *perf->mutable_op() = BuildOpInfoWithoutDevice(node, name_to_node, inputs)
[all...] |
| graph_properties.cc | 1128 std::unordered_map<string, const NodeDef*> name_to_node; \/\/ Empty local
[all...] |
| virtual_scheduler.cc | 335 // TODO(dyoon): this is a bit inefficient as name_to_node is already built in
341 std::unordered_map<string, const NodeDef*> name_to_node; local
343 name_to_node[node->name()] = node;
380 const NodeDef* input_node = name_to_node[NodeName(input_node_name)];
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session.h | 122 // 'graph' is the entire graph being executed. 'name_to_node'
123 // maps node name to node. We keep 'graph' and 'name_to_node' only in
133 NameNodeMap name_to_node; member in struct:tensorflow::DirectSession::ExecutorsAndKeys
|
| step_stats_collector.cc | 193 std::unordered_map<StringPiece, Node*, StringPieceHasher> name_to_node; local
195 name_to_node.emplace(n->name(), n);
225 const Node* node = name_to_node[stats.node_name()];
|
| direct_session.cc | 1008 const NameNodeMap* name_to_node = &executors_and_keys->name_to_node; local
[all...] |
