提交 5e4160d8 编写于 作者: Y yangfei

imp dropout op kernel for gpu

上级 7dbb4fb8
...@@ -25,9 +25,7 @@ bool MulKernel<GPU_CL, float>::Init(MulParam<GPU_CL> *param) { ...@@ -25,9 +25,7 @@ bool MulKernel<GPU_CL, float>::Init(MulParam<GPU_CL> *param) {
} }
template <> template <>
void MulKernel<GPU_CL, float>::Compute(const MulParam<GPU_CL> &param) { void MulKernel<GPU_CL, float>::Compute(const MulParam<GPU_CL> &param) {}
}
template class MulKernel<GPU_CL, float>; template class MulKernel<GPU_CL, float>;
......
...@@ -58,6 +58,9 @@ namespace ops = paddle_mobile::operators; ...@@ -58,6 +58,9 @@ namespace ops = paddle_mobile::operators;
#ifdef PADDLE_MOBILE_CPU #ifdef PADDLE_MOBILE_CPU
REGISTER_OPERATOR_CPU(mul, ops::MulOp); REGISTER_OPERATOR_CPU(mul, ops::MulOp);
#endif #endif
#ifdef PADDLE_MOBILE_CL
REGISTER_OPERATOR_CL(mul, ops::MulOp);
#endif
#ifdef PADDLE_MOBILE_MALI_GPU #ifdef PADDLE_MOBILE_MALI_GPU
REGISTER_OPERATOR_MALI_GPU(mul, ops::MulOp); REGISTER_OPERATOR_MALI_GPU(mul, ops::MulOp);
#endif #endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册