diff --git a/lite/backends/arm/math/conv3x3_winograd_fp32_c4.cc b/lite/backends/arm/math/conv3x3_winograd_fp32_c4.cc index 67d60b18141f64fd4e0048e1a5d1e2c5373c7484..69a2454cf0a5d08e1db64b8ce7b95821e393a0eb 100644 --- a/lite/backends/arm/math/conv3x3_winograd_fp32_c4.cc +++ b/lite/backends/arm/math/conv3x3_winograd_fp32_c4.cc @@ -79,6 +79,7 @@ void conv_compute_6x6_3x3(const float* input, const float* bias, const operators::ConvParam& param, ARMContext* ctx) { + auto act_param = param.activation_param; const int pad_h = (*param.paddings)[0]; const int pad_w = (*param.paddings)[2]; float* tmp_work_space = @@ -296,7 +297,7 @@ void conv_compute_6x6_3x3(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } else { for (int ci = 0; ci < oc_4; ++ci) { @@ -343,7 +344,7 @@ void conv_compute_6x6_3x3(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } } @@ -366,6 +367,7 @@ void conv_compute_2x2_3x3(const float* input, const float* bias, const operators::ConvParam& param, ARMContext* ctx) { + auto act_param = param.activation_param; const int pad_h = (*param.paddings)[0]; const int pad_w = (*param.paddings)[2]; float* tmp_work_space = @@ -565,7 +567,7 @@ void conv_compute_2x2_3x3(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } else { for (int ci = 0; ci < oc_4; ++ci) { @@ -606,7 +608,7 @@ void conv_compute_2x2_3x3(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } } @@ -627,6 +629,7 @@ void conv_compute_2x2_3x3_small(const float* input, const float* bias, const operators::ConvParam& param, ARMContext* ctx) { + auto act_param = param.activation_param; const int pad_h = (*param.paddings)[0]; const int pad_w = (*param.paddings)[2]; float* tmp_work_space = @@ -819,7 +822,7 @@ void conv_compute_2x2_3x3_small(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } else { for (int ci = 0; ci < oc_4; ++ci) { @@ -860,7 +863,7 @@ void conv_compute_2x2_3x3_small(const float* input, wout, false, zero_ptr, - nullptr); + &act_param); } } }