diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/convolver.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/convolver.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/convolver.hpp b/src/core/NEON/kernels/arm_gemm/convolver.hpp index b15f669132..a9f3dd0ad8 100644 --- a/src/core/NEON/kernels/arm_gemm/convolver.hpp +++ b/src/core/NEON/kernels/arm_gemm/convolver.hpp @@ -231,8 +231,8 @@ public: for (unsigned int ky=0; ky<params.kernel_height; ky++) { for (unsigned int kx=0; kx<params.kernel_width; kx++) { unsigned int n = (ky * params.kernel_width) + kx; - m_kernel_y[n] = ky - params.padding_top; - m_kernel_x[n] = kx - params.padding_left; + m_kernel_y[n] = (ky * params.dilation_h) - params.padding_top; + m_kernel_x[n] = (kx * params.dilation_w) - params.padding_left; } } } |