diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp')
-rw-r--r-- | arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp b/arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp index 62a20c9eea..6e06db324c 100644 --- a/arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp +++ b/arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp @@ -65,11 +65,11 @@ inline void BlockedGemm( const int c_row_stride ) { // Array access methods - const auto A = [a, M, K, a_row_stride] (const int i, const int j) -> TIn { + const auto A = [a, a_row_stride] (const int i, const int j) -> TIn { return a[i*a_row_stride + j]; }; - const auto B = [b, K, N, b_row_stride] (const int i, const int j) -> TIn { + const auto B = [b, b_row_stride] (const int i, const int j) -> TIn { return b[i*b_row_stride + j]; }; |