提交 fa0a708d 编写于 作者: C chenjiaoAngel

add 5x5s1_dw

上级 ce77f797
此差异已折叠。
...@@ -141,7 +141,24 @@ void conv_depthwise_5x5s1_fp32(float* dout, ...@@ -141,7 +141,24 @@ void conv_depthwise_5x5s1_fp32(float* dout,
int padh, int padh,
const operators::ConvParam& param, const operators::ConvParam& param,
ARMContext* ctx); 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, void conv_depthwise_5x5s2_fp32(const float* din,
float* dout, float* dout,
int num, int num,
......
...@@ -751,6 +751,7 @@ void conv_depthwise_5x5_fp32(const void* din, ...@@ -751,6 +751,7 @@ void conv_depthwise_5x5_fp32(const void* din,
act_param, act_param,
ctx); ctx);
} else if (stride == 1) { } else if (stride == 1) {
#if 1
conv_depthwise_5x5s1_fp32(reinterpret_cast<float*>(dout), conv_depthwise_5x5s1_fp32(reinterpret_cast<float*>(dout),
reinterpret_cast<const float*>(din), reinterpret_cast<const float*>(din),
reinterpret_cast<const float*>(weights), reinterpret_cast<const float*>(weights),
...@@ -767,6 +768,26 @@ void conv_depthwise_5x5_fp32(const void* din, ...@@ -767,6 +768,26 @@ void conv_depthwise_5x5_fp32(const void* din,
pad_h, pad_h,
param, param,
ctx); 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 { } else {
LOG(FATAL) << "unsupport this type 5x5 dw conv"; 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.
先完成此消息的编辑!
想要评论请 注册