Home
last modified time | relevance | path

Searched defs:bernoulli_tensor_kernel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistributionBernoulli.cu26 void bernoulli_tensor_kernel(const TensorBase &self, const TensorBase &p_, std::optional<Generator>… in bernoulli_tensor_kernel() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistributionKernels.cpp34 void bernoulli_tensor_kernel(const TensorBase &self, const TensorBase &p_, std::optional<Generator>… in bernoulli_tensor_kernel() function