Home
last modified time | relevance | path

Searched refs:base_constraint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dconstraints.py227 def __init__(self, base_constraint, reinterpreted_batch_ndims): argument
228 assert isinstance(base_constraint, Constraint)
231 self.base_constraint = base_constraint
237 return self.base_constraint.is_discrete
241 return self.base_constraint.event_dim + self.reinterpreted_batch_ndims
244 result = self.base_constraint.check(value)
246 expected = self.base_constraint.event_dim + self.reinterpreted_batch_ndims
H A Dconstraint_registry.py171 base_transform = biject_to(constraint.base_constraint)
179 base_transform = transform_to(constraint.base_constraint)
/aosp_15_r20/external/pytorch/test/distributions/
H A Dtest_transforms.py153 domain = domain.base_constraint