Searched refs:UniformDistribution (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op_gpu.cu.cc | 71 GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >;
73 GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >;
75 GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >;
77 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int32> >;
79 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >;
|
| D | random_op.cc | 248 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Compute()
307 typedef random::UniformDistribution<PhiloxRandom, double> Uniform; in Compute()
458 CPUDevice, random::UniformDistribution<random::PhiloxRandom, TYPE>>; \
470 PhiloxRandomOp<CPUDevice, random::UniformDistribution< \
494 CPUDevice, random::UniformDistribution<random::PhiloxRandom, IntType>>; \
522 PhiloxRandomOp<GPUDevice, random::UniformDistribution< \
545 GPUDevice, random::UniformDistribution<random::PhiloxRandom, IntType>>; \
709 SYCLDevice, random::UniformDistribution<random::PhiloxRandom, TYPE>>; \
715 PhiloxRandomOp<SYCLDevice, random::UniformDistribution< \
|
| D | stateless_random_ops.cc | 141 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Fill()
160 StatelessRandomOp<DEVICE##Device, random::UniformDistribution< \
|
| D | multinomial_op_gpu.cu.cc | 66 typedef random::UniformDistribution<random::PhiloxRandom, float> Dist; in operator ()()
|
| D | parameterized_truncated_normal_op.cc | 83 typedef random::UniformDistribution<random::PhiloxRandom, T> Uniform; in operator ()()
293 random::UniformDistribution<random::PhiloxRandom, T>::kElementCost; in operator ()()
|
| D | random_poisson_op.cc | 103 typedef random::UniformDistribution<random::PhiloxRandom, CT> Uniform; in operator ()()
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 64 typedef random::UniformDistribution<random::PhiloxRandom, T> Uniform; in TruncatedNormalKernel()
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | random_distributions.h | 70 class UniformDistribution; variable
73 class UniformDistribution<Generator, Eigen::half> {
97 class UniformDistribution<Generator, bfloat16> {
121 class UniformDistribution<Generator, float> {
145 class UniformDistribution<Generator, double> {
169 class UniformDistribution<Generator, int32> {
182 UniformDistribution(int32 lo, int32 hi) in UniformDistribution() function
204 class UniformDistribution<Generator, int64> {
217 UniformDistribution(int64 lo, int64 hi) in UniformDistribution() function
|
| D | random_distributions_test.cc | 157 FillRandoms<UniformDistribution<PhiloxRandom, T> >(gen, &v1[0], v1.size()); in UniformMomentsTest()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_random_uniform.cc | 30 typedef tensorflow::random::UniformDistribution< in ComputeRandomUniformArray()
|