Searched refs:bernoulli_tensor_kernel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionBernoulli.cu | 26 void bernoulli_tensor_kernel(const TensorBase &self, const TensorBase &p_, std::optional<Generator>… in bernoulli_tensor_kernel() function
37 REGISTER_DISPATCH(bernoulli_tensor_stub, &bernoulli_tensor_kernel);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DistributionTemplates.h | 362 template<template<typename> class bernoulli_tensor_kernel, typename RNG>
367 bernoulli_tensor_kernel<RNG>()(self, p_, gen); in bernoulli_impl_()
380 template<template<typename> class bernoulli_tensor_kernel, typename RNG>
386 bernoulli_impl_<bernoulli_tensor_kernel, RNG>(result, self, gen); in bernoulli_out_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 34 void bernoulli_tensor_kernel(const TensorBase &self, const TensorBase &p_, std::optional<Generator>… in bernoulli_tensor_kernel() function
238 REGISTER_DISPATCH(bernoulli_tensor_stub, &bernoulli_tensor_kernel);
|