| 120 | } |
| 121 | |
| 122 | bool Graph::remove_connection(EdgeID eid) |
| 123 | { |
| 124 | if (eid >= _edges.size()) |
| 125 | { |
| 126 | return false; |
| 127 | } |
| 128 | |
| 129 | std::unique_ptr<Edge> &edge = _edges[eid]; |
| 130 | |
| 131 | // Remove node connections |
| 132 | if (edge != nullptr) |
| 133 | { |
| 134 | // Get tensor bound to the edge |
| 135 | if (edge->tensor() != nullptr) |
| 136 | { |
| 137 | edge->tensor()->unbind_edge(eid); |
| 138 | } |
| 139 | |
| 140 | // Remove edges from source node |
| 141 | if (edge->producer() != nullptr) |
| 142 | { |
| 143 | edge->producer()->_output_edges.erase(eid); |
| 144 | } |
| 145 | |
| 146 | // Remove edges from sink node |
| 147 | if ((edge->consumer() != nullptr) && (edge->consumer_idx() < edge->consumer()->_input_edges.size())) |
| 148 | { |
| 149 | edge->consumer()->_input_edges[edge->consumer_idx()] = EmptyEdgeID; |
| 150 | } |
| 151 | } |
| 152 | |
| 153 | // Clear edge |
| 154 | edge = nullptr; |
| 155 | |
| 156 | return true; |
| 157 | } |
| 158 | |
| 159 | TensorID Graph::create_tensor(const TensorDescriptor &desc) |
| 160 | { |
nothing calls this directly
no test coverage detected