diff options
| -rw-r--r-- | src/core/NEON/kernels/winograd/gemm.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/winograd/gemm.hpp b/src/core/NEON/kernels/winograd/gemm.hpp index 564016a646..111e19602a 100644 --- a/src/core/NEON/kernels/winograd/gemm.hpp +++ b/src/core/NEON/kernels/winograd/gemm.hpp @@ -65,11 +65,11 @@ 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]; }; |