提交 2b3e3621 编写于 作者: D dongzhihong

"template specialization link include"

上级 30a47fe8
...@@ -15,6 +15,21 @@ bool Gaussian(DeviceContext& ctx, ...@@ -15,6 +15,21 @@ bool Gaussian(DeviceContext& ctx,
const T& std, const T& std,
const T& seed); const T& seed);
template <typename T>
bool Gaussian(platform::CPUDeviceContext& ctx,
framework::Tensor* output,
const int size,
const T& mean,
const T& std,
const T& seed);
template <typename T>
bool Gaussian(platform::CUDADeviceContext& ctx,
framework::Tensor* output,
const int size,
const T& mean,
const T& std,
const T& seed);
template <typename Place, typename T> template <typename Place, typename T>
class RandomOpKernel : public framework::OpKernel { class RandomOpKernel : public framework::OpKernel {
public: public:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册