| 66 | } |
| 67 | |
| 68 | void NETensorHandle::release_if_unused() |
| 69 | { |
| 70 | // TODO (geopin01): Release tensor only if all sub-tensors are marked as not used |
| 71 | if (!_tensor.is_used()) |
| 72 | { |
| 73 | _tensor.allocator()->free(); |
| 74 | } |
| 75 | } |
| 76 | |
| 77 | const arm_compute::ITensor &NETensorHandle::tensor() const |
| 78 | { |