From 748c06d7e644e2b9f93074fdcfbe57a3eb7ee9d0 Mon Sep 17 00:00:00 2001 From: zhangyang Date: Fri, 10 Aug 2018 16:17:56 +0800 Subject: [PATCH] register Feed and Fetch ops --- src/operators/feed_op.h | 1 + src/operators/fetch_op.cpp | 1 + src/operators/fetch_op.h | 1 + 3 files changed, 3 insertions(+) diff --git a/src/operators/feed_op.h b/src/operators/feed_op.h index b34c7cf78b..4766d56d9a 100644 --- 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 diff --git a/src/operators/fetch_op.cpp b/src/operators/fetch_op.cpp index adbd61d5ec..30cddceaa4 100644 --- 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 diff --git a/src/operators/fetch_op.h b/src/operators/fetch_op.h index 5614fef8fe..417637c800 100644 --- 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 -- GitLab