diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-10-01 15:39:42 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-10-01 15:31:55 +0000 |
commit | 20c2b501f206e4db1c15c2334f9e8a1baf640a50 (patch) | |
tree | 0cf99d218353d49a8a19076725a12652b8cfd4c4 /src/runtime/NEON/functions/NEFullyConnectedLayer.cpp | |
parent | b94c368a39ba6518d843c66a156f825a2c66af60 (diff) | |
download | ComputeLibrary-20c2b501f206e4db1c15c2334f9e8a1baf640a50.tar.gz |
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>
Diffstat (limited to 'src/runtime/NEON/functions/NEFullyConnectedLayer.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEFullyConnectedLayer.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp index 7adc3bca9e..ee622f4699 100644 --- a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp +++ b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp @@ -426,7 +426,6 @@ void NEFullyConnectedLayer::prepare() { 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 @@ void NEFullyConnectedLayer::prepare() { _converted_weights_output.allocator()->allocate(); _convert_weights.run(); + cur_weights->mark_as_unused(); } - cur_weights->mark_as_unused(); _are_weights_converted = true; } |