Fork自 PaddlePaddle / Paddle
体验新版 GitCode,发现更多精彩内容 >>
#define EIGEN_USE_GPU
#include "paddle/operators/sigmoid_op.h"
REGISTER_OP_GPU_KERNEL(sigmoid, ops::SigmoidKernel<ops::GPUPlace, float>);