sgd_op.cu 106 字节
Newer Older
Q
Qiao Longfei 已提交
1 2
#include "paddle/operators/sgd_op.h"

3
REGISTER_OP_GPU_KERNEL(sgd, ops::SGDOpKernel<ops::GPUPlace, float>);