Searched defs:standard_uniform (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Distributions.cpp | 83 at::uniform_real_distribution<double> standard_uniform(0.0, 1.0); in sample_poisson() local
439 at::uniform_real_distribution<double> standard_uniform(0.0, 1.0); in _s_binomial_cpu() local
442 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_binomial_cpu() local
480 at::uniform_real_distribution<double> standard_uniform(0.0, 1.0); in _s_gamma_cpu() local
483 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_gamma_cpu() local
513 at::uniform_real_distribution<double> standard_uniform(0.0, 1.0); in _s_dirichlet_cpu() local
516 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_dirichlet_cpu() local
|
| D | Distributions.h | 90 …r_t sample_gamma(scalar_t alpha, BaseSampler<accscalar_t, uniform_sampler_t>& standard_uniform, Ba… in sample_gamma()
151 …ion(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, uniform_sampler_t>& standard_uniform) { in binomial_inversion()
171 …trs(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, uniform_sampler_t>& standard_uniform) { in btrs()
224 …ial(scalar_t count, scalar_t prob, BaseSampler<accscalar_t, uniform_sampler_t>& standard_uniform) { in sample_binomial()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Distributions.cu | 114 BaseSampler<accscalar_t, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in gamma_cuda_kernel() local
|