diff options
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h b/arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h index b7b2587454..c2bb47c550 100644 --- a/arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h +++ b/arm_compute/runtime/CL/functions/CLLocallyConnectedLayer.h @@ -90,6 +90,7 @@ public: // Inherited methods overridden: void run() override; + void prepare() override; private: CLMemoryGroup _memory_group; @@ -100,7 +101,7 @@ private: CLTensor _input_im2col_reshaped; CLTensor _weights_reshaped; CLTensor _gemm_output; - bool _is_first_run; + bool _is_prepared; const ICLTensor *_original_weights; }; } |