Searched refs:bernoulli_scalar_kernel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DistributionBernoulli.cu | 31 void bernoulli_scalar_kernel(const TensorBase &self, double p, std::optional<Generator> gen) { in bernoulli_scalar_kernel() function
38 REGISTER_DISPATCH(bernoulli_scalar_stub, &bernoulli_scalar_kernel);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | DistributionKernels.cpp | 45 void bernoulli_scalar_kernel(const TensorBase &self, double p, std::optional<Generator> gen) { in bernoulli_scalar_kernel() function
49 void bernoulli_scalar_kernel(const TensorBase &self, double p, std::optional<Generator> gen) { in bernoulli_scalar_kernel() function
239 REGISTER_DISPATCH(bernoulli_scalar_stub, &bernoulli_scalar_kernel);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DistributionTemplates.h | 371 template<template<typename> class bernoulli_scalar_kernel, typename RNG>
376 bernoulli_scalar_kernel<RNG>()(self, p, gen); in bernoulli_impl_()
|