Searched refs:FillPhiloxRandom (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_gpu.cu.cc | 53 void FillPhiloxRandom<GPUDevice, Distribution>::operator()( in operator ()()
70 template struct FillPhiloxRandom< variable
72 template struct FillPhiloxRandom< variable
74 template struct FillPhiloxRandom< variable
76 template struct FillPhiloxRandom< variable
78 template struct FillPhiloxRandom< variable
80 template struct FillPhiloxRandom< variable
82 template struct FillPhiloxRandom< variable
84 template struct FillPhiloxRandom< variable
86 template struct FillPhiloxRandom< variable
[all …]
|
| D | random_op.h | 29 struct FillPhiloxRandom;
38 struct FillPhiloxRandom<CPUDevice, Distribution> {
49 struct FillPhiloxRandom<GPUDevice, Distribution> {
61 struct FillPhiloxRandom<SYCLDevice, Distribution> {
|
| D | random_op.cc | 63 struct FillPhiloxRandom { struct
154 void FillPhiloxRandom<CPUDevice, Distribution>::operator()( in operator ()()
204 functor::FillPhiloxRandom<Device, Distribution>()( in Compute()
253 functor::FillPhiloxRandom<Device, Distribution>()( in Compute()
457 template struct functor::FillPhiloxRandom< \
459 template struct functor::FillPhiloxRandom< \
461 template struct functor::FillPhiloxRandom< \
493 template struct functor::FillPhiloxRandom< \
544 template struct functor::FillPhiloxRandom< \
683 void FillPhiloxRandom<SYCLDevice, Distribution>::operator()( in operator ()()
[all …]
|
| D | stateless_random_ops.cc | 110 functor::FillPhiloxRandom<Device, Distribution>()( in Fill()
147 functor::FillPhiloxRandom<Device, Distribution>()( in Fill()
|
| D | multinomial_op_gpu.cu.cc | 67 functor::FillPhiloxRandom<GPUDevice, Dist>()(ctx, d, gen, noises.data(), in operator ()()
|
| D | stateful_random_ops.cc | 38 functor::FillPhiloxRandom<CPUDevice, Distribution>()( in operator ()()
|