| 82 | } // namespace |
| 83 | |
| 84 | void TFOp::AddNode(TFGraphNode* node) { |
| 85 | const string& op = node->op(); |
| 86 | if (tfcnodes_map_.find(op) == tfcnodes_map_.end()) { |
| 87 | tfcnodes_map_[op] = |
| 88 | std::unique_ptr<TFMultiGraphNode>(new TFMultiGraphNode(op)); |
| 89 | } |
| 90 | TFMultiGraphNode* tfcnode = tfcnodes_map_[op].get(); |
| 91 | tfcnode->AddGraphNode(node); |
| 92 | } |
| 93 | |
| 94 | void TFOp::Build() { |
| 95 | for (auto& tn : tfcnodes_map_) { |