sigmoid_op.cu 117 字节
Newer Older
1
#include "paddle/operators/sigmoid_op.h"
2

3
REGISTER_OP_GPU_KERNEL(sigmoid, ops::SigmoidKernel<ops::GPUPlace, float>);