diff options
Diffstat (limited to 'tensorflow/core/kernels/random_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/random_op_gpu.cu.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/random_op_gpu.cu.cc b/tensorflow/core/kernels/random_op_gpu.cu.cc index 9c51708c4c..e1e5c16a79 100644 --- a/tensorflow/core/kernels/random_op_gpu.cu.cc +++ b/tensorflow/core/kernels/random_op_gpu.cu.cc @@ -145,6 +145,8 @@ struct FillPhiloxRandom<GPUDevice, Distribution> { // clang-format off // NVCC cannot handle ">>" properly template struct FillPhiloxRandom< + GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >; +template struct FillPhiloxRandom< GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >; template struct FillPhiloxRandom< GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >; @@ -153,11 +155,16 @@ template struct FillPhiloxRandom< template struct FillPhiloxRandom< GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >; template struct FillPhiloxRandom< + GPUDevice, random::NormalDistribution<random::PhiloxRandom, Eigen::half> >; +template struct FillPhiloxRandom< GPUDevice, random::NormalDistribution<random::PhiloxRandom, float> >; template struct FillPhiloxRandom< GPUDevice, random::NormalDistribution<random::PhiloxRandom, double> >; template struct FillPhiloxRandom< GPUDevice, random::TruncatedNormalDistribution< + random::SingleSampleAdapter<random::PhiloxRandom>, Eigen::half> >; +template struct FillPhiloxRandom< + GPUDevice, random::TruncatedNormalDistribution< random::SingleSampleAdapter<random::PhiloxRandom>, float> >; template struct FillPhiloxRandom< GPUDevice, random::TruncatedNormalDistribution< |