From 2b3e362136f94a20503cdb1133762e33b18bf6f3 Mon Sep 17 00:00:00 2001 From: dongzhihong Date: Tue, 25 Jul 2017 15:50:51 +0800 Subject: [PATCH] "template specialization link include" --- paddle/operators/random_op.h | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/paddle/operators/random_op.h b/paddle/operators/random_op.h index bee6cc9cbd..3eeb1f87c8 100644 --- a/paddle/operators/random_op.h +++ b/paddle/operators/random_op.h @@ -15,6 +15,21 @@ bool Gaussian(DeviceContext& ctx, const T& std, const T& seed); +template +bool Gaussian(platform::CPUDeviceContext& ctx, + framework::Tensor* output, + const int size, + const T& mean, + const T& std, + const T& seed); +template +bool Gaussian(platform::CUDADeviceContext& ctx, + framework::Tensor* output, + const int size, + const T& mean, + const T& std, + const T& seed); + template class RandomOpKernel : public framework::OpKernel { public: