Searched refs:uniform_and_transform (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionTemplates.h | 402 void uniform_and_transform(TensorIteratorBase& iter, RNG gen, transform_t transform) { in uniform_and_transform() function
477 uniform_and_transform<scalar_t, opmath_t>(iter, gen, uniform_func); in uniform_kernel()
521 uniform_and_transform<scalar_t, accscalar_t>(iter, gen, geometric_func); in geometric_kernel()
544 uniform_and_transform<scalar_t, accscalar_t>(iter, gen, exponential_func); in exponential_kernel()
567 uniform_and_transform<scalar_t, accscalar_t>(iter, gen, cauchy_func); in cauchy_kernel()
657 uniform_and_transform<scalar_t, accscalar_t>(iter, gen, bernoulli_func); in bernoulli_kernel()
|