sigmoid_op.cu 191 字节
Newer Older
1 2 3 4
#include <paddle/operators/sigmoid_op.h>
#include <paddle/framework/op_registry.h>

REGISTER_OP_GPU_KERNEL(
Q
qijun 已提交
5
    sigmoid, paddle::operators::SigmoidKernel<paddle::platform::GPUPlace, float>);