Searched refs:uniform_lambda (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Distributions.cu | 85 auto uniform_lambda = curand_uniform_wrapper(state); in binomial_cuda_kernel()
86 BaseSampler<accscalar_t, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in binomial_cuda_kernel()
87 …auto sample = sample_binomial<scalar_t, accscalar_t, decltype(uniform_lambda)>(count, prob, standa… in binomial_cuda_kernel()
111 auto uniform_lambda = [&state] __device__ () { in gamma_cuda_kernel()
114 BaseSampler<accscalar_t, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in gamma_cuda_kernel()
120 …auto sample = sample_gamma<scalar_t, accscalar_t, decltype(uniform_lambda), decltype(normal_lambda… in gamma_cuda_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distributions.cpp | 438 auto uniform_lambda = [generator] () { in _s_binomial_cpu() local
442 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_binomial_cpu()
444 …auto sample = sample_binomial<scalar_t, double, decltype(uniform_lambda)>(count_val, prob_val, sta… in _s_binomial_cpu()
479 auto uniform_lambda = [generator] () { in _s_gamma_cpu() local
483 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_gamma_cpu()
490 …auto sample = sample_gamma<scalar_t, double, decltype(uniform_lambda), decltype(normal_lambda)>(al… in _s_gamma_cpu()
512 auto uniform_lambda = [generator] () { in _s_dirichlet_cpu() local
516 BaseSampler<double, decltype(uniform_lambda)> standard_uniform(uniform_lambda); in _s_dirichlet_cpu()
523 auto sample = sample_gamma<double, double, decltype(uniform_lambda), decltype(normal_lambda)> in _s_dirichlet_cpu()
|