Free _gemm_output_3d's memory region after run()

Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I689a9b6ad5a4486316363674491c594e153b19fb

diff --git a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
index f3e7e197332b858d707379328df4b30cdf9e840c..2d296f39ea108c93a6d929b0420955e78e6fbe13 100644 (file)
--- a/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEGEMMConvolutionLayer.cpp
@@ -611,6 +611,8 @@ void NEGEMMConvolutionLayer::run()
     {
         _reshape_layer.run();
     }
+
+    _gemm_output_3d.allocator()->free();
 }
 
 void NEGEMMConvolutionLayer::prepare()