Home
last modified time | relevance | path

Searched refs:corr_cholesky (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributions/
H A Dtest_constraints.py79 (constraints.corr_cholesky,),
126 if constraint_fn is constraints.corr_cholesky:
163 if constraint_fn is constraints.corr_cholesky:
H A Dtest_transforms.py188 elif domain is constraints.corr_cholesky:
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dconstraint_registry.py263 @biject_to.register(constraints.corr_cholesky)
264 @transform_to.register(constraints.corr_cholesky)
H A Dlkj_cholesky.py61 support = constraints.corr_cholesky
H A Dconstraints.py675 corr_cholesky = _CorrCholesky() variable
H A Dtransforms.py828 codomain = constraints.corr_cholesky