Paddle/paddle/operators/sgd_op.cu

3 lines
106 B

#include "paddle/operators/sgd_op.h"
REGISTER_OP_GPU_KERNEL(sgd, ops::SGDOpKernel<ops::GPUPlace, float>);