Fork自 PaddlePaddle / Paddle
#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, RandomOpKernel_GPU_float);