Home
last modified time | relevance | path

Searched refs:nodes_map_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc69 auto node = nodes_map_.find(v.first); in TFStats()
70 if (node != nodes_map_.end()) { in TFStats()
99 new TFGraphNode(node_pb.second, profile, &id_to_string_, &nodes_map_)); in TFStats()
100 nodes_map_.insert(std::pair<string, std::unique_ptr<TFGraphNode>>( in TFStats()
112 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in BuildView()
119 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in BuildView()
126 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in BuildView()
133 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in BuildView()
199 if (nodes_map_.find(node.name()) != nodes_map_.end()) { in AddGraph()
203 size_t num_nodes = nodes_map_.size(); in AddGraph()
[all …]
Dtfprof_scope.cc32 if (nodes_map_.find(name) != nodes_map_.end()) { in CreateParentNode()
33 return nodes_map_[name].get(); in CreateParentNode()
40 nodes_map_[name] = in CreateParentNode()
42 return nodes_map_[name].get(); in CreateParentNode()
47 if (nodes_map_.find(node->name()) == nodes_map_.end()) { in AddNode()
48 nodes_map_[name] = std::unique_ptr<ScopeNode>(new ScopeNode(node)); in AddNode()
54 if (nodes_map_.find(name) == nodes_map_.end()) { in AddNode()
66 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in Build()
73 nodes_map_[prefix]->children.push_back(node); in Build()
Dtfprof_graph.cc35 nodes_map_[name] = in CreateParentNode()
37 return nodes_map_[name].get(); in CreateParentNode()
42 nodes_map_[name] = std::unique_ptr<GraphNode>(new GraphNode(node)); in AddNode()
50 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in Build()
56 auto child_it = nodes_map_.find(inputs_it->second); in Build()
57 if (child_it != nodes_map_.end()) { in Build()
63 for (auto it = nodes_map_.begin(); it != nodes_map_.end(); it++) { in Build()
Dtfprof_stats.h64 return nodes_map_; in nodes()
69 return covered_nodes_.size() / (nodes_map_.size() + 1e-10); in run_coverage()
115 std::map<string, std::unique_ptr<TFGraphNode>> nodes_map_; variable
Dtfprof_node.h312 nodes_map_ = nodes_map; in TFGraphNode()
318 nodes_map_ = nodes_map; in TFGraphNode()
709 if (!nodes_map_) continue; in input_shapes()
710 auto input_it = nodes_map_->find(inp.second); in input_shapes()
711 if (input_it == nodes_map_->end()) continue; in input_shapes()
731 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map_;
Dtfprof_scope.h72 std::map<string, std::unique_ptr<ScopeNode>> nodes_map_; variable
Dtfprof_graph.h83 std::map<string, std::unique_ptr<GraphNode>> nodes_map_; variable