|
|
|
@ -80,4 +80,4 @@ Use to initialize tensor with gaussian random generator.
|
|
|
|
|
namespace ops = paddle::operators;
|
|
|
|
|
REGISTER_OP_WITHOUT_GRADIENT(gaussian_random, ops::GaussianRandomOp,
|
|
|
|
|
ops::GaussianRandomOpMaker);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(gaussian_random, ops::CPUGaussianRandomKernel<float>);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(gaussian_random, ops::CPUGaussianRandomKernel<float>);
|
|
|
|
|