| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_distributions.py | 1507 def ref_log_prob(idx, val, log_prob): argument
1574 def ref_log_prob(idx, val, log_prob): argument
1635 def ref_log_prob(idx, x, log_prob): argument
1739 def ref_log_prob(idx, x, log_prob): argument
1877 def ref_log_prob(idx, val, log_prob): argument
1955 def ref_log_prob(ref_rate, idx, x, log_prob): argument
2286 def ref_log_prob(idx, x, log_prob): argument
2372 def ref_log_prob(idx, x, log_prob): argument
2528 def ref_log_prob(idx, x, log_prob): argument
2611 def ref_log_prob(idx, x, log_prob): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | adaptive.cpp | 194 Tensor AdaptiveLogSoftmaxWithLossImpl::AdaptiveLogSoftmaxWithLossImpl::log_prob( in log_prob() function in torch::nn::AdaptiveLogSoftmaxWithLossImpl::AdaptiveLogSoftmaxWithLossImpl
209 const Tensor log_prob = _get_full_log_prob(input, head_output); in predict() local
212 const Tensor log_prob = _get_full_log_prob( in predict() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ctc_decoder_ops.cc | 61 Tensor** log_prob, OpOutputList* decoded_indices, in ValidateInputsGenerateOutputs()
196 Tensor* log_prob = nullptr; in Compute() local
301 Tensor* log_prob = nullptr; in Compute() local
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | lkj_cholesky.py | 118 def log_prob(self, value): member in LKJCholesky
|
| H A D | fishersnedecor.py | 92 def log_prob(self, value): member in FisherSnedecor
|
| H A D | gumbel.py | 60 def log_prob(self, value): member in Gumbel
|
| H A D | poisson.py | 68 def log_prob(self, value): member in Poisson
|
| H A D | studentT.py | 93 def log_prob(self, value): member in StudentT
|
| H A D | geometric.py | 118 def log_prob(self, value): member in Geometric
|
| H A D | half_cauchy.py | 65 def log_prob(self, value): member in HalfCauchy
|
| H A D | half_normal.py | 60 def log_prob(self, value): member in HalfNormal
|
| H A D | transformed_distribution.py | 158 def log_prob(self, value): member in TransformedDistribution
|
| H A D | cauchy.py | 75 def log_prob(self, value): member in Cauchy
|
| H A D | laplace.py | 80 def log_prob(self, value): member in Laplace
|
| H A D | multinomial.py | 128 def log_prob(self, value): member in Multinomial
|
| H A D | uniform.py | 84 def log_prob(self, value): member in Uniform
|
| H A D | beta.py | 80 def log_prob(self, value): member in Beta
|
| H A D | negative_binomial.py | 115 def log_prob(self, value): member in NegativeBinomial
|
| H A D | mixture_same_family.py | 161 def log_prob(self, x): member in MixtureSameFamily
|
| H A D | independent.py | 109 def log_prob(self, value): member in Independent
|
| H A D | gamma.py | 82 def log_prob(self, value): member in Gamma
|
| H A D | exponential.py | 66 def log_prob(self, value): member in Exponential
|
| H A D | categorical.py | 137 def log_prob(self, value): member in Categorical
|
| H A D | dirichlet.py | 79 def log_prob(self, value): member in Dirichlet
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | adaptive.py | 279 def log_prob(self, input: Tensor) -> Tensor: member in AdaptiveLogSoftmaxWithLoss
|