| 27 | namespace tensorflow { |
| 28 | namespace tfprof { |
| 29 | GraphNode* TFGraph::CreateParentNode(const string& name) { |
| 30 | node_defs_.push_back(std::unique_ptr<NodeDef>(new NodeDef())); |
| 31 | node_defs_.back()->set_name(name); |
| 32 | node_defs_.back()->set_op(kTFGraphParent); |
| 33 | parent_nodes_[name] = std::unique_ptr<TFGraphNode>( |
| 34 | new TFGraphNode(node_defs_.back().get(), -1, nullptr)); |
| 35 | nodes_map_[name] = |
| 36 | std::unique_ptr<GraphNode>(new GraphNode(parent_nodes_[name].get())); |
| 37 | return nodes_map_[name].get(); |
| 38 | } |
| 39 | |
| 40 | void TFGraph::AddNode(TFGraphNode* node) { |
| 41 | string name = node->name(); |