提交 79081309 编写于 作者: C chenjiaoAngel

fix v8 compute errot

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