sgd_op.cu 230 字节
Newer Older
Q
Qiao Longfei 已提交
1 2 3 4 5
#include "paddle/operators/sgd_op.h"
#include "paddle/framework/op_registry.h"

typedef paddle::operators::SGDOpKernel<::paddle::platform::GPUPlace, float> SGDOpKernel_GPU_float;
REGISTER_OP_GPU_KERNEL(sgd, SGDOpKernel_GPU_float);