diff options
Diffstat (limited to 'arm_compute/runtime/NEON')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h index fe65ac1a43..84d3594426 100644 --- a/arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDepthwiseConvolutionLayer.h @@ -82,6 +82,8 @@ private: bool _is_quantized; bool _is_optimized; bool _are_weights_reshaped; + bool _is_nchw; + bool _is_first_run; }; /** Basic function to execute a generic depthwise convolution. This function calls the following NEON kernels: |