diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h index 55921f78f3..c1260977c0 100644 --- a/arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h @@ -74,6 +74,7 @@ public: // Inherited methods overridden: void run() override; + void prepare() override; /** Static function to check if given info will lead to a valid configuration of @ref NEGEMMConvolutionLayer * @@ -122,7 +123,7 @@ private: const ITensor *_input; const ITensor *_weights; ITensor *_output; - bool _reshaped_kernel; + bool _is_prepared; bool _is_activationlayer_enabled; }; } |