aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp')
-rw-r--r--arm_compute/core/NEON/kernels/convolution/winograd/gemm.hpp4
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];
};