From 8ecfd853b18202fd83ff7e6deca28f2e392f9aa2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= <liyinhgqw@gmail.com>
Date: Thu, 19 Apr 2018 20:10:23 +0800
Subject: [PATCH] Fix conv1x1 pad

---
 mace/kernels/arm/conv_2d.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mace/kernels/arm/conv_2d.cc b/mace/kernels/arm/conv_2d.cc
index 3b99890e..6bf952e2 100644
--- a/mace/kernels/arm/conv_2d.cc
+++ b/mace/kernels/arm/conv_2d.cc
@@ -362,14 +362,14 @@ void Conv2dFunctor<DeviceType::NEON, float>::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) {
-- 
GitLab