提交 015a408a 编写于 作者: C chenjiaoAngel

fix ut

上级 0559b0a6
......@@ -223,7 +223,7 @@ void test_conv_fp32(const std::vector<DDim>& input_dims,
leakey_relu_scale);
if (flag_act == 2) { // relu6
for (int i = 0; i < dim_out.production(); i++) {
dout_basic_fp32[i] = dout_basic[i] > six ? six : dout_basic[i];
dout_basic[i] = dout_basic[i] > six ? six : dout_basic[i];
}
}
}
......
......@@ -563,7 +563,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) {
for (auto& cout : {1, 5, 17}) {
for (auto& g : {1, 2}) {
for (auto& flag_bias : {false, true}) {
for (auto& flag_act : {0, 1, 2, 4}) {
for (auto& flag_act : {0, 1, 2}) {
std::vector<DDim> dims;
if (cin % g != 0 || cout % g != 0) {
continue;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册