"template specialization link include"

revert-3824-remove_grad_op_type
dongzhihong 8 years ago
parent 30a47fe8a3
commit 2b3e362136

@ -15,6 +15,21 @@ bool Gaussian(DeviceContext& ctx,
const T& std,
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>
class RandomOpKernel : public framework::OpKernel {
public:

Loading…
Cancel
Save