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