Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dmultivariate_normal.py137 scale_tril=None, argument
209 def scale_tril(self): member in MultivariateNormal
H A Dlowrank_multivariate_normal.py160 def scale_tril(self): member in LowRankMultivariateNormal
H A Dwishart.py181 def scale_tril(self): member in Wishart
/aosp_15_r20/external/pytorch/test/distributions/
H A Dtest_distributions.py2898 mean, covariance=None, precision=None, scale_tril=None argument
2906 def gradcheck_func(samples, mu, sigma, prec, scale_tril): argument
3123 df=None, covariance=None, precision=None, scale_tril=None argument
3129 def gradcheck_func(samples, nu, sigma, prec, scale_tril): argument