Searched defs:log_normal_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionLogNormalKernel.cu | 8 void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, std::optional<Generator> … in log_normal_kernel() function
|
| H A D | DistributionTemplates.h | 491 void log_normal_kernel(TensorIteratorBase& iter, double mean_, double std_, RNG gen) { in log_normal_kernel() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 203 static void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, std::optional<Gene… in log_normal_kernel() function
|
| H A D | DistributionTemplates.h | 309 void log_normal_kernel(TensorIteratorBase& iter, double mean, double std, RNG generator) { in log_normal_kernel() function
|