From fed98beba549d665d2a4b49dcb8509bc03295d80 Mon Sep 17 00:00:00 2001 From: Liangliang He Date: Wed, 27 Sep 2017 17:27:48 +0800 Subject: [PATCH] Fix conv_2d 1x1 test breakage --- mace/kernels/neon/conv_2d_neon_1x1.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/kernels/neon/conv_2d_neon_1x1.cc b/mace/kernels/neon/conv_2d_neon_1x1.cc index c1010bcd..e3fd325b 100644 --- a/mace/kernels/neon/conv_2d_neon_1x1.cc +++ b/mace/kernels/neon/conv_2d_neon_1x1.cc @@ -38,7 +38,7 @@ void Conv2dNeonK1x1S1(const float *input, // NCHW #pragma omp parallel for collapse(2) for (index_t n = 0; n < batch; ++n) { for (index_t c = 0; c < channels; ++c) { - const float *filter_ptr = filter; + const float *filter_ptr = filter + c * input_channels; // TODO Will GCC opt these out? float *channel_output_start = output + n * channels * height * width + c * height * width; -- GitLab