| 96 | const GraphDef& GetGraphDef() const { return graph_def_; } |
| 97 | |
| 98 | void Extend(const SubGraph &sub_graph) { |
| 99 | for (const Node* node : sub_graph.GetNodes()) { |
| 100 | if (loc_ == "") { |
| 101 | loc_ = sub_graph.GetLoc(); |
| 102 | } |
| 103 | AddNode(node); |
| 104 | } |
| 105 | } |
| 106 | |
| 107 | void SetGraphHandle(const std::string &handle) { |
| 108 | graph_handle_ = handle; |
no test coverage detected