Fork自 PaddlePaddle / Paddle
#include <paddle/operators/sigmoid_op.h> #include <paddle/framework/op_registry.h> REGISTER_OP_GPU_KERNEL( sigmoid, paddle::operators::SigmoidKernel<paddle::platform::GPUPlace>);