提交 3eee4b8b 编写于 作者: Z zhaojiaying01

add fetch kernel for FPGA, GPU_MALI

上级 47bca6fc
...@@ -19,16 +19,16 @@ namespace paddle_mobile { ...@@ -19,16 +19,16 @@ namespace paddle_mobile {
namespace operators { namespace operators {
template <> template <>
bool FetchKernel<CPU, float>::Init(FetchParam<CPU> *param) { bool FetchKernel<FPGA, float>::Init(FetchParam<FPGA> *param) {
return true; return true;
} }
template <> template <>
void FetchKernel<CPU, float>::Compute(const FetchParam<CPU> &param) { void FetchKernel<FPGA, float>::Compute(const FetchParam<FPGA> &param) {
param.Out()->ShareDataWith(*(param.InputX())); param.Out()->ShareDataWith(*(param.InputX()));
} }
template class FetchKernel<CPU, float>; template class FetchKernel<FPGA, float>;
} // namespace operators } // namespace operators
} // namespace paddle_mobile } // namespace paddle_mobile
......
...@@ -19,16 +19,16 @@ namespace paddle_mobile { ...@@ -19,16 +19,16 @@ namespace paddle_mobile {
namespace operators { namespace operators {
template <> template <>
bool FetchKernel<CPU, float>::Init(FetchParam<CPU> *param) { bool FetchKernel<GPU_MALI, float>::Init(FetchParam<GPU_MALI> *param) {
return true; return true;
} }
template <> template <>
void FetchKernel<CPU, float>::Compute(const FetchParam<CPU> &param) { void FetchKernel<GPU_MALI, float>::Compute(const FetchParam<GPU_MALI> &param) {
param.Out()->ShareDataWith(*(param.InputX())); param.Out()->ShareDataWith(*(param.InputX()));
} }
template class FetchKernel<CPU, float>; template class FetchKernel<GPU_MALI, float>;
} // namespace operators } // namespace operators
} // namespace paddle_mobile } // namespace paddle_mobile
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册