fix cpu conv_add and conv_add_relu (#1654)

* fix cpu conv_add and conv_add_relu

* fix cpu conv_add and conv_add_relu

src/operators/kernel/arm/convolution/conv_add_kernel.cpp

@@ -52,7 +52,13 @@ void ConvAddKernel<CPU, float>::Compute(const FusionConvAddParam<CPU> &param) {
       PADDLE_MOBILE_THROW_EXCEPTION("Invalid convolution execute mode %d",
                                     param.ExecMode());
   }
-  math::AddChannelWise<IDENTITY>(param.Output(), param.Bias(), param.Output());
+  if (param.Bias()->dims() == param.Output()->dims()) {
+    math::AddElememtWise<IDENTITY>(param.Output(), param.Bias(), param.Axis(),
+                                   param.Output());
+  } else {
+    math::AddChannelWise<IDENTITY>(param.Output(), param.Bias(),
+                                   param.Output());
+  }
 }
 
 template class ConvAddKernel<CPU, float>;

src/operators/kernel/arm/convolution/conv_add_relu_kernel.cpp

@@ -53,7 +53,12 @@ void ConvAddReluKernel<CPU, float>::Compute(
       PADDLE_MOBILE_THROW_EXCEPTION("Invalid convolution execute mode %d",
                                     param.ExecMode());
   }
-  math::AddChannelWise<RELU>(param.Output(), param.Bias(), param.Output());
+  if (param.Bias()->dims() == param.Output()->dims()) {
+    math::AddElememtWise<RELU>(param.Output(), param.Bias(), param.Axis(),
+                               param.Output());
+  } else {
+    math::AddChannelWise<RELU>(param.Output(), param.Bias(), param.Output());
+  }
 }
 
 template class ConvAddReluKernel<CPU, float>;