sigmoid_op.cu 223 字节
Newer Older
Q
qijun 已提交
1
#define EIGEN_USE_GPU
2
#include "paddle/operators/sigmoid_op.h"
3

4
REGISTER_OP_GPU_KERNEL(sigmoid, ops::SigmoidKernel<ops::GPUPlace, float>);
5
REGISTER_OP_GPU_KERNEL(sigmoid_grad, ops::SigmoidGradKernel<ops::GPUPlace, float>);