| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | relaxed_categorical.py | 37 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in ExpRelaxedCategorical
118 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in RelaxedOneHotCategorical
|
| H A D | relaxed_bernoulli.py | 41 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in LogitRelaxedBernoulli
130 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in RelaxedBernoulli
|
| H A D | chi2.py | 24 arg_constraints = {"df": constraints.positive} variable in Chi2
|
| H A D | logistic_normal.py | 34 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in LogisticNormal
|
| H A D | lkj_cholesky.py | 60 arg_constraints = {"concentration": constraints.positive} variable in LKJCholesky
|
| H A D | weibull.py | 29 arg_constraints = { variable in Weibull
|
| H A D | pareto.py | 27 arg_constraints = {"alpha": constraints.positive, "scale": constraints.positive} variable in Pareto
|
| H A D | kumaraswamy.py | 40 arg_constraints = { variable in Kumaraswamy
|
| H A D | fishersnedecor.py | 31 arg_constraints = {"df1": constraints.positive, "df2": constraints.positive} variable in FisherSnedecor
|
| H A D | gumbel.py | 31 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Gumbel
|
| H A D | log_normal.py | 30 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in LogNormal
|
| H A D | inverse_gamma.py | 33 arg_constraints = { variable in InverseGamma
|
| H A D | poisson.py | 32 arg_constraints = {"rate": constraints.nonnegative} variable in Poisson
|
| H A D | studentT.py | 32 arg_constraints = { variable in StudentT
|
| H A D | geometric.py | 44 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in Geometric
|
| H A D | half_cauchy.py | 32 arg_constraints = {"scale": constraints.positive} variable in HalfCauchy
|
| H A D | half_normal.py | 32 arg_constraints = {"scale": constraints.positive} variable in HalfNormal
|
| H A D | transformed_distribution.py | 49 arg_constraints: Dict[str, constraints.Constraint] = {} variable in TransformedDistribution
|
| H A D | cauchy.py | 33 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Cauchy
|
| H A D | laplace.py | 29 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Laplace
|
| H A D | multinomial.py | 50 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in Multinomial
|
| H A D | uniform.py | 32 arg_constraints = { variable in Uniform
|
| H A D | beta.py | 32 arg_constraints = { variable in Beta
|
| H A D | negative_binomial.py | 31 arg_constraints = { variable in NegativeBinomial
|
| H A D | mixture_same_family.py | 54 arg_constraints: Dict[str, constraints.Constraint] = {} variable in MixtureSameFamily
|