未验证 提交 a9d17eef 编写于 作者: X xiaogang 提交者: GitHub

[ARM] fix 5x5s2_int8 compute bug, test=develop (#3079)

上级 1ae03f55
......@@ -102,7 +102,7 @@ void conv_depthwise_5x5s2_int8(Dtype* dout,
if (h + hout_r_block > hout) {
h_kernel = hout - h;
}
int hs = h - padh;
int hs = h * 2 - padh;
int he = hs + h_kernel * 2 + 3;
#pragma omp parallel for num_threads(threads)
......
......@@ -502,7 +502,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) {
std::vector<DDim> dims;
DDim weights_dim({c, 1, 5, 5});
for (auto& batch : {1, 2}) {
for (auto& h : {1, 3, 15, 33}) {
for (auto& h : {1, 3, 15, 33, 112, 224}) {
dims.push_back(DDim({batch, c, h, h}));
}
}
......@@ -514,7 +514,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) {
{1, 1},
flag_bias,
flag_relu,
{4},
{1, 4},
{FLAGS_power_mode});
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册