Searched defs:uniform_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionUniform.cu | 8 void uniform_kernel(TensorIteratorBase& iter, double from, double to, std::optional<Generator> gen)… in uniform_kernel() function
|
| H A D | DistributionTemplates.h | 458 void uniform_kernel(TensorIteratorBase& iter, double from_, double to_, RNG gen) { in uniform_kernel() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 208 void uniform_kernel(TensorIteratorBase& iter, double from, double to, std::optional<Generator> gen)… in uniform_kernel() function
|
| H A D | DistributionTemplates.h | 267 void uniform_kernel(TensorIteratorBase& iter, double from_, double to_, RNG generator) { in uniform_kernel() function
|