aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2017-09-12 19:19:12 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit898a806bd0c72aeafb3557efdbc686aab54992d9 (patch)
tree74bb2693a372e89e5eee7616c476e34d4f221ce4 /arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h
parent14e7d14d2109d32be9c132fa85dc5b70cf5e3a07 (diff)
downloadComputeLibrary-898a806bd0c72aeafb3557efdbc686aab54992d9.tar.gz
COMPMID-417: Fix NEDirectConvolutionLayer
Change-Id: I62a1fc7253a4f597d0d63b80310e0c84c3602b1a Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87436 Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h')
-rw-r--r--arm_compute/core/NEON/kernels/NEDirectConvolutionLayerKernel.h1
1 files changed, 1 insertions, 0 deletions
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;
};