diff --git a/mace/kernels/arm/conv_2d.cc b/mace/kernels/arm/conv_2d.cc index 3b99890ea7aa85646ec14fd8d73f739af7894724..6bf952e222ac6738b8564cc940ce782b5eb6468b 100644 --- a/mace/kernels/arm/conv_2d.cc +++ b/mace/kernels/arm/conv_2d.cc @@ -362,14 +362,14 @@ void Conv2dFunctor::operator()(const Tensor *input, }; } else if (use_neon_1x1_s1) { conv_func = [=](const float *pad_input, float *pad_output) { - Conv2dNeonK1x1S1(input_data, + Conv2dNeonK1x1S1(pad_input, filter_data, batch, - height, - width, + extra_input_height, + extra_input_width, input_channels, channels, - output_data); + pad_output); }; } else { conv_func = [=](const float *pad_input, float *pad_output) {