commit 20c2b501f206e4db1c15c2334f9e8a1baf640a50
author Michalis Spyrou <michalis.spyrou@arm.com> Tue Oct 01 15:39:42 2019 +0100
committer Michalis Spyrou <michalis.spyrou@arm.com> Tue Oct 01 15:31:55 2019 +0000
tree 0cf99d218353d49a8a19076725a12652b8cfd4c4
parent b94c368a39ba6518d843c66a156f825a2c66af60

COMPMID-2693 Weights manager to mark the initial weights as unsued

Change-Id: Iaa964e71d0aa80fff0a465127ac3716563bbe85f
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2017
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

src/runtime/NEON/functions/NEFullyConnectedLayer.cpp

diff --git a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp
index 7adc3bc..ee622f4 100644
--- a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp
+++ b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp
@@ -426,7 +426,6 @@
         {
             if(_weights_manager && _weights_manager->are_weights_managed(_original_weights))
             {
-                cur_weights->mark_as_unused();
                 cur_weights = _weights_manager->run(cur_weights, &_reshape_weights_managed_function);
             }
             else
@@ -455,9 +454,9 @@
             {
                 _converted_weights_output.allocator()->allocate();
                 _convert_weights.run();
+                cur_weights->mark_as_unused();
             }
 
-            cur_weights->mark_as_unused();
             _are_weights_converted = true;
         }