提交 79081309 编写于 作者: C chenjiaoAngel

fix v8 compute errot

上级 3e8bfc7b
...@@ -753,7 +753,7 @@ void conv_depthwise_5x5_fp32(const void* din, ...@@ -753,7 +753,7 @@ void conv_depthwise_5x5_fp32(const void* din,
param, param,
act_param, act_param,
ctx); ctx);
else { } else {
conv_depthwise_5x5s2_fp32(reinterpret_cast<float*>(dout), conv_depthwise_5x5s2_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),
......
...@@ -60,7 +60,7 @@ void DepthwiseConv<PRECISION(kFloat), PRECISION(kFloat)>::PrepareForRun() { ...@@ -60,7 +60,7 @@ void DepthwiseConv<PRECISION(kFloat), PRECISION(kFloat)>::PrepareForRun() {
auto strides = param.strides; auto strides = param.strides;
bool pads_five = (paddings[0] < 5) || (paddings[2] < 5); bool pads_five = (paddings[0] < 5) || (paddings[2] < 5);
if (ch_four && pads_five && win >= 2 * kw && hin >= kw && if (ch_four && pads_five && win >= 2 * kw && hin >= kw &&
(strides[0] == 2 && strides[1] == 2) { (strides[0] == 2 && strides[1] == 2)) {
flag_trans_weights_ = false; flag_trans_weights_ = false;
impl_ = lite::arm::math::conv_depthwise_5x5_fp32; impl_ = lite::arm::math::conv_depthwise_5x5_fp32;
#ifdef LITE_WITH_PROFILE #ifdef LITE_WITH_PROFILE
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册