Searched refs:FillPhiloxRandomKernel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_gpu.h | 29 struct FillPhiloxRandomKernel;
32 struct FillPhiloxRandomKernel<Distribution, false> {
39 struct FillPhiloxRandomKernel<Distribution, true> {
131 PHILOX_DEVICE_FUNC void FillPhiloxRandomKernel<Distribution, false>::Run(
162 PHILOX_DEVICE_FUNC void FillPhiloxRandomKernel<Distribution, true>::Run(
|
| D | random_op.cc | 573 struct FillPhiloxRandomKernel;
576 struct FillPhiloxRandomKernel<Distribution, false> { struct
581 FillPhiloxRandomKernel(write_accessor& data, random::PhiloxRandom& gen, in FillPhiloxRandomKernel() function
623 struct FillPhiloxRandomKernel<Distribution, true> { struct
628 FillPhiloxRandomKernel(write_accessor& data, random::PhiloxRandom& gen, in FillPhiloxRandomKernel() argument
695 FillPhiloxRandomKernel<Distribution, in operator ()()
|
| D | random_op_gpu.cu.cc | 46 FillPhiloxRandomKernel<Distribution, in FillPhiloxRandomKernelLaunch()
|
| D | stateful_random_ops_gpu.cu.cc | 43 functor::FillPhiloxRandomKernel<Distribution, in FillKernel()
|