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