arm_compute v17.05

[platform/upstream/armcl.git] / src / runtime / TensorAllocator.cpp

diff --git a/src/runtime/TensorAllocator.cpp b/src/runtime/TensorAllocator.cpp
index a272ac2..5c719c7 100644 (file)
--- a/src/runtime/TensorAllocator.cpp
+++ b/src/runtime/TensorAllocator.cpp
@@ -101,6 +101,14 @@ void TensorAllocator::allocate()
     info().set_is_resizable(false);
 }
 
+void TensorAllocator::free()
+{
+    ARM_COMPUTE_ERROR_ON(_buffer == nullptr);
+
+    _buffer.reset();
+    info().set_is_resizable(true);
+}
+
 uint8_t *TensorAllocator::lock()
 {
     return (_buffer != nullptr) ? _buffer.get()->data() : nullptr;