From 898a806bd0c72aeafb3557efdbc686aab54992d9 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 12 Sep 2017 19:19:12 +0100 Subject: COMPMID-417: Fix NEDirectConvolutionLayer Change-Id: I62a1fc7253a4f597d0d63b80310e0c84c3602b1a Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87436 Tested-by: Kaizen Reviewed-by: Moritz Pflanzer --- arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h') diff --git a/arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h b/arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h index 4de6c12218..928ac3569d 100644 --- a/arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h @@ -74,6 +74,7 @@ private: PadStrideInfo _conv_info; BorderSize _border_size; unsigned int _kernel_size; + unsigned int _num_weight_elems_read_per_row; unsigned int _num_elems_read_per_iteration; unsigned int _num_elems_written_per_iteration; }; -- cgit v1.2.1