Home
last modified time | relevance | path

Searched defs:random_kernel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistributionRandomKernel.cu18 void random_kernel(TensorIteratorBase& iter, std::optional<Generator> gen_) { in random_kernel() function
H A DDistributionTemplates.h363 void random_kernel(TensorIteratorBase& iter, RNG gen) { in random_kernel() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistributionKernels.cpp223 static void random_kernel(TensorIteratorBase& iter, std::optional<Generator> gen) { in random_kernel() function
H A DDistributionTemplates.h68 void random_kernel(TensorIteratorBase& iter, RNG generator) { in random_kernel() function