Searched defs:random_from_to_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionRandomKernel.cu | 8 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, std::optional<Ge… in random_from_to_kernel() function
|
| H A D | DistributionTemplates.h | 287 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, RNG gen) { in random_from_to_kernel() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 218 static void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, std::opti… in random_from_to_kernel() function
|
| H A D | DistributionTemplates.h | 26 void random_from_to_kernel(TensorIteratorBase& iter, uint64_t range, int64_t base, RNG generator) { in random_from_to_kernel() function
|