register Feed and Fetch ops

748c06d7 · zhangyang · 2af8e6dc · 748c06d7 · 748c06d7 · 748c06d7
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

src/operators/feed_op.h src/operators/feed_op.h +1 -0

src/operators/fetch_op.cpp src/operators/fetch_op.cpp +1 -0

src/operators/fetch_op.h src/operators/fetch_op.h +1 -0

未找到文件。
--- a/src/operators/feed_op.h
+++ b/src/operators/feed_op.h
@@ -74,4 +74,5 @@ USE_OP_CPU(feed);
 USE_OP_MALI_GPU(feed);
 #endif
 #ifdef PADDLE_MOBILE_FPGA
+USE_OP_FPGA(feed);
 #endif
--- a/src/operators/fetch_op.cpp
+++ b/src/operators/fetch_op.cpp
@@ -25,4 +25,5 @@ REGISTER_OPERATOR_CPU(fetch, ops::FetchOp);
 REGISTER_OPERATOR_MALI_GPU(fetch, ops::FetchOp);
 #endif
 #ifdef PADDLE_MOBILE_FPGA
+REGISTER_OPERATOR_FPGA(fetch, ops::FetchOp);
 #endif
--- a/src/operators/fetch_op.h
+++ b/src/operators/fetch_op.h
@@ -54,4 +54,5 @@ USE_OP_CPU(fetch);
 USE_OP_MALI_GPU(fetch);
 #endif
 #ifdef PADDLE_MOBILE_FPGA
+USE_OP_FPGA(fetch);
 #endif