| 363 | } |
| 364 | |
| 365 | bool Graph::RemoveTensor(Tensor* tensor) |
| 366 | { |
| 367 | /* remove all refers from consumer */ |
| 368 | |
| 369 | for(unsigned int i = 0; i < tensor->consumer.size(); i++) |
| 370 | { |
| 371 | NodePort* port = tensor->consumer[i]; |
| 372 | Node* node = port->owner; |
| 373 | int idx = port->port_index; |
| 374 | |
| 375 | node->RemoveInputPort(idx); |
| 376 | } |
| 377 | |
| 378 | /* remove refer from producer */ |
| 379 | |
| 380 | NodePort* port = tensor->producer; |
| 381 | |
| 382 | if(port) |
| 383 | { |
| 384 | Node* node = port->owner; |
| 385 | int idx = port->port_index; |
| 386 | |
| 387 | node->RemoveOutputPort(idx); |
| 388 | } |
| 389 | |
| 390 | /* release the tensor*/ |
| 391 | |
| 392 | auto ir = tensor_map_.find(tensor->GetName()); |
| 393 | |
| 394 | if(ir != tensor_map_.end()) |
| 395 | tensor_map_.erase(ir); |
| 396 | |
| 397 | /* if it is my tensor */ |
| 398 | if(RemoveTensorOwner(tensor)) |
| 399 | delete tensor; |
| 400 | |
| 401 | return true; |
| 402 | } |
| 403 | |
| 404 | bool Graph::RemoveNode(Node* node) |
| 405 | { |