diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-04-05 01:56:05 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-05 03:03:21 -0700 |
commit | 287589e38097407e1f7b14efce01120284b208a5 (patch) | |
tree | 73f77aff054eb9f9d3fc6cd54f8b22822e42e7de /tensorflow/core/kernels/random_op_gpu.cu.cc | |
parent | bc507335f3c72f7ab070785c8f10ea0e19af6209 (diff) |
Implement random number generation ops for Eigen::half.
Change: 119030756
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< |