Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dunary_test.py35 def bernoulli_(input): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistributions.cpp174 Tensor& bernoulli_(Tensor& self, const Tensor& p_, std::optional<Generator> gen) { in bernoulli_() function
178 Tensor& bernoulli_(Tensor& self, double p, std::optional<Generator> gen) { in bernoulli_() function
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions_for_rng.py252 def bernoulli_(self, p=0.5): function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py1888 def bernoulli_(x, *args): function