提交 1053795d 编写于 作者: X xiaogang 提交者: GitHub

feat: winograd support relu6 (#2781)

上级 04833837
......@@ -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);
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册