提交 fa0a708d 编写于 作者: C chenjiaoAngel

add 5x5s1_dw

上级 ce77f797
此差异已折叠。
......@@ -141,7 +141,24 @@ void conv_depthwise_5x5s1_fp32(float* dout,
int padh,
const operators::ConvParam& param,
ARMContext* ctx);
void conv_depthwise_5x5s1_fp32(float* dout,
const float* din,
const float* weights,
const float* bias,
bool flag_bias,
bool flag_relu,
int num,
int chin,
int hin,
int win,
int hout,
int wout,
int pad_top,
int pad_bottom,
int pad_left,
int pad_right,
const operators::ActivationParam& act_param,
ARMContext* ctx);
void conv_depthwise_5x5s2_fp32(const float* din,
float* dout,
int num,
......
......@@ -751,6 +751,7 @@ void conv_depthwise_5x5_fp32(const void* din,
act_param,
ctx);
} else if (stride == 1) {
#if 1
conv_depthwise_5x5s1_fp32(reinterpret_cast<float*>(dout),
reinterpret_cast<const float*>(din),
reinterpret_cast<const float*>(weights),
......@@ -767,6 +768,26 @@ void conv_depthwise_5x5_fp32(const void* din,
pad_h,
param,
ctx);
#else
conv_depthwise_5x5s1_fp32(reinterpret_cast<float*>(dout),
reinterpret_cast<const float*>(din),
reinterpret_cast<const float*>(weights),
bias,
flag_bias,
flag_relu,
num,
ch_in,
h_in,
w_in,
h_out,
w_out,
paddings[0],
paddings[1],
paddings[2],
paddings[3],
act_param,
ctx);
#endif
} else {
LOG(FATAL) << "unsupport this type 5x5 dw conv";
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册