提交 beb53b38 编写于 作者: H hong19860320

change pool to pool2d to fix the unit test of pooling arm kernel

test=develop
上级 c07b215d
......@@ -261,8 +261,8 @@ TEST(pool_arm, compute) {
}
TEST(pool, retrive_op) {
auto pool =
KernelRegistry::Global().Create<TARGET(kARM), PRECISION(kFloat)>("pool");
auto pool = KernelRegistry::Global().Create<TARGET(kARM), PRECISION(kFloat)>(
"pool2d");
ASSERT_FALSE(pool.empty());
ASSERT_TRUE(pool.front());
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册