diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/ITensorPack.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/arm_compute/core/ITensorPack.h b/arm_compute/core/ITensorPack.h index 2f41d4d51e..17b7241862 100644 --- a/arm_compute/core/ITensorPack.h +++ b/arm_compute/core/ITensorPack.h @@ -95,6 +95,11 @@ public: * @return The pointer to the tensor if exist and is const else nullptr */ const ITensor *get_const_tensor(int id) const; + /** Remove the tensor stored with the given id + * + * @param[in] id ID of tensor to remove + */ + void remove_tensor(int id); /** Pack size accessor * * @return Number of tensors registered to the pack |