Skip to content

Mace
Mace

Xiaomi / Mace

通知 106
Star 40
Fork 27
Mace
Mace

体验新版 GitCode，发现更多精彩内容 >>

提交 096289e8 编写于 作者: U Unknown 提交者： liutuo
浏览文件
操作

fix scalar_math benchmark bug 

not support max and min yet
上级 3b300782
隐藏空白更改
内联 并排
Showing with 6 addition and 7 deletion
mace/ops/scalar_math_benchmark.cc
浏览文件 @ 096289e8
...@@ -71,17 +71,16 @@ static void ScalarMath(int iters, int batch, int channels, ...@@ -71,17 +71,16 @@ static void ScalarMath(int iters, int batch, int channels,
BM_SCALAR_MATH(1, 1, 512, 512, 2, 0); BM_SCALAR_MATH(1, 1, 512, 512, 2, 0);
BM_SCALAR_MATH(1, 3, 128, 128, 2, 1); BM_SCALAR_MATH(1, 3, 128, 128, 2, 1);
BM_SCALAR_MATH(1, 3, 512, 512, 2, 2); BM_SCALAR_MATH(1, 3, 512, 512, 2, 4);
BM_SCALAR_MATH(1, 32, 112, 112, 2, 3); BM_SCALAR_MATH(1, 32, 112, 112, 2, 5);
BM_SCALAR_MATH(1, 64, 256, 256, 3, 0); BM_SCALAR_MATH(1, 64, 256, 256, 3, 0);
BM_SCALAR_MATH(1, 64, 512, 512, 3, 1); BM_SCALAR_MATH(1, 64, 512, 512, 3, 1);
BM_SCALAR_MATH(1, 128, 56, 56, 3, 2); BM_SCALAR_MATH(1, 128, 56, 56, 3, 4);
BM_SCALAR_MATH(1, 128, 256, 256, 3, 3); BM_SCALAR_MATH(1, 128, 256, 256, 3, 5);
BM_SCALAR_MATH(1, 256, 14, 14, 3, 0); BM_SCALAR_MATH(1, 256, 14, 14, 3, 0);
BM_SCALAR_MATH(1, 512, 14, 14, 3, 1); BM_SCALAR_MATH(1, 512, 14, 14, 3, 1);
BM_SCALAR_MATH(1, 1024, 7, 7, 3, 2); BM_SCALAR_MATH(1, 1024, 7, 7, 3, 4);
BM_SCALAR_MATH(32, 1, 256, 256, 3, 3); BM_SCALAR_MATH(32, 1, 256, 256, 3, 5);
BM_SCALAR_MATH(32, 3, 256, 256, 3, 2);
} // namespace test } // namespace test
} // namespace ops } // namespace ops
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或