random_op.cu 248 字节
Newer Older
D
dongzhihong 已提交
1 2 3 4 5 6
#include "paddle/operators/random_op.h"
#include "paddle/framework/op_registry.h"

typedef paddle::operators::RandomOpKernel<paddle::platform::GPUPlace, float>
  RandomOpKernel_GPU_float;
REGISTER_OP_GPU_KERNEL(random_op, RandomOpKernel_GPU_float);