Searched refs:normal_lambda (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Distributions.cu | 116 auto normal_lambda = [&state] __device__ () { in gamma_cuda_kernel()
119 BaseSampler<accscalar_t, decltype(normal_lambda)> standard_normal(normal_lambda); in gamma_cuda_kernel()
120 …ample_gamma<scalar_t, accscalar_t, decltype(uniform_lambda), decltype(normal_lambda)>(alpha, stand… in gamma_cuda_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distributions.cpp | 485 auto normal_lambda = [generator] () { in _s_gamma_cpu() local
489 BaseSampler<double, decltype(normal_lambda)> standard_normal(normal_lambda); in _s_gamma_cpu()
490 …auto sample = sample_gamma<scalar_t, double, decltype(uniform_lambda), decltype(normal_lambda)>(al… in _s_gamma_cpu()
518 auto normal_lambda = [generator] () { in _s_dirichlet_cpu() local
522 BaseSampler<double, decltype(normal_lambda)> standard_normal(normal_lambda); in _s_dirichlet_cpu()
523 auto sample = sample_gamma<double, double, decltype(uniform_lambda), decltype(normal_lambda)> in _s_dirichlet_cpu()
|