Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistributionGeometricKernel.cu8 void geometric_kernel(TensorIteratorBase& iter, double p_, std::optional<Generator> gen) { in geometric_kernel() function
H A DDistributionTemplates.h514 void geometric_kernel(TensorIteratorBase& iter, double p, RNG gen) { in geometric_kernel() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistributionKernels.cpp198 static void geometric_kernel(TensorIteratorBase& iter, double p, std::optional<Generator> gen) { in geometric_kernel() function
H A DDistributionTemplates.h329 void geometric_kernel(TensorIteratorBase& iter, double p, RNG generator) { in geometric_kernel() function