Home
last modified time | relevance | path

Searched defs:log_prob (Results 26 – 41 of 41) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dbinomial.py125 def log_prob(self, value): member in Binomial
H A Dmultivariate_normal.py247 def log_prob(self, value): member in MultivariateNormal
H A Dbernoulli.py109 def log_prob(self, value): member in Bernoulli
H A Dlowrank_multivariate_normal.py213 def log_prob(self, value): member in LowRankMultivariateNormal
H A Dnormal.py80 def log_prob(self, value): member in Normal
H A Dvon_mises.py134 def log_prob(self, value): member in VonMises
H A Done_hot_categorical.py101 def log_prob(self, value): member in OneHotCategorical
H A Drelaxed_categorical.py85 def log_prob(self, value): member in ExpRelaxedCategorical
H A Drelaxed_bernoulli.py102 def log_prob(self, value): member in LogitRelaxedBernoulli
H A Dwishart.py291 def log_prob(self, value): member in Wishart
H A Dcontinuous_bernoulli.py176 def log_prob(self, value): member in ContinuousBernoulli
H A Ddistribution.py188 def log_prob(self, value: torch.Tensor) -> torch.Tensor: member in Distribution
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dintegration.cpp183 auto log_prob = p[action].log(); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/
H A Ddistribution.py777 def log_prob(self, value, name="log_prob"): member in Distribution
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py2040 def log_prob(x): function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h3838 const float log_prob = scale * input_data[j] - precomputed; in LogSoftmax() local

12