提交 400f599c 编写于 作者: H hong19860320

change pool to pool2d

test=develop
上级 33437fe4
...@@ -272,4 +272,4 @@ TEST(pool, retrive_op) { ...@@ -272,4 +272,4 @@ TEST(pool, retrive_op) {
} // namespace lite } // namespace lite
} // namespace paddle } // namespace paddle
USE_LITE_KERNEL(pool, kARM, kFloat, kNCHW, def); USE_LITE_KERNEL(pool2d, kARM, kFloat, kNCHW, def);
...@@ -86,5 +86,5 @@ TEST(pool_op_lite, test) { ...@@ -86,5 +86,5 @@ TEST(pool_op_lite, test) {
} // namespace paddle } // namespace paddle
#ifdef LITE_WITH_ARM #ifdef LITE_WITH_ARM
USE_LITE_KERNEL(pool, kARM, kFloat, kNCHW, def); USE_LITE_KERNEL(pool2d, kARM, kFloat, kNCHW, def);
#endif #endif
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册