diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-08-22 14:28:30 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | ba1ffe96eb4563ba7e18b39728d9db373c62f7c3 (patch) | |
tree | 0e931ef35b271f353d98aec59f9e3042471b4aea /src/runtime/CL/functions/CLGEMM.cpp | |
parent | 3ada2b7a29e1ab2058ab7dc701cacff548d2aae9 (diff) | |
download | ComputeLibrary-ba1ffe96eb4563ba7e18b39728d9db373c62f7c3.tar.gz |
COMPMID-1537: Fix weights retention in CLFullyConnectedLayer
Change-Id: Id978c34889b86fa8b9184d3349cc9b12837141a2
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145403
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLGEMM.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLGEMM.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLGEMM.cpp b/src/runtime/CL/functions/CLGEMM.cpp index 1ad8531920..85d90a05e1 100644 --- a/src/runtime/CL/functions/CLGEMM.cpp +++ b/src/runtime/CL/functions/CLGEMM.cpp @@ -86,7 +86,7 @@ void CLGEMM::configure(const ICLTensor *a, const ICLTensor *b, const ICLTensor * // Check if we need to reshape the matrix B only on the first run _reshape_b_only_on_first_run = gemm_info.reshape_b_only_on_first_run(); - _is_prepared = false; + _is_prepared = gemm_info.retain_internal_weights(); _original_b = b; const ICLTensor *matrix_a = a; |