Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 52eb86e3 编写于 作者: T tensor-tang
浏览文件
操作

refine im2col benchmark

上级 3017f460
显示空白变更内容
内联 并排
Showing with 7 addition and 10 deletion
paddle/fluid/operators/math/im2col_test.cc
浏览文件 @ 52eb86e3
......@@ -256,14 +256,11 @@ TEST(math, im2col_cputest) {
/*pw*/ 2);
// benchmark
LOG(INFO) << "padding == 0";
benchIm2col(/*ic*/ 3, /*ih*/ 224, /*iw*/ 224, /*fh*/ 3, /*fw*/ 3, /*ph*/ 0,
/*pw*/ 0);
benchIm2col(/*ic*/ 3, /*ih*/ 224, /*iw*/ 224, /*fh*/ 5, /*fw*/ 5, /*ph*/ 0,
/*pw*/ 0);
LOG(INFO) << "padding == 1";
benchIm2col(/*ic*/ 3, /*ih*/ 224, /*iw*/ 224, /*fh*/ 3, /*fw*/ 3, /*ph*/ 1,
/*pw*/ 1);
benchIm2col(/*ic*/ 3, /*ih*/ 224, /*iw*/ 224, /*fh*/ 5, /*fw*/ 5, /*ph*/ 1,
/*pw*/ 1);
for (int p : {0, 1, 2}) {
for (int k : {3, 5}) {
LOG(INFO) << "padding == " << p << ", filter == " << k;
benchIm2col(/*ic*/ 3, /*ih*/ 224, /*iw*/ 224, /*fh*/ k, /*fw*/ k,
/*ph*/ p, /*pw*/ p);
}
}
}
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或