Home
last modified time | relevance | path

Searched refs:enumerate_support (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dindependent.py117 def enumerate_support(self, expand=True): member in Independent
122 return self.base_dist.enumerate_support(expand=expand)
H A Dbinomial.py152 log_prob = self.log_prob(self.enumerate_support(False))
155 def enumerate_support(self, expand=True): member in Binomial
H A Dbernoulli.py120 def enumerate_support(self, expand=True): member in Bernoulli
H A Done_hot_categorical.py110 def enumerate_support(self, expand=True): member in OneHotCategorical
H A Dmultinomial.py121 support = self._binomial.enumerate_support(expand=False)[1:]
H A Dcategorical.py151 def enumerate_support(self, expand=True): member in Categorical
H A Ddistribution.py218 def enumerate_support(self, expand: bool = True) -> torch.Tensor: member in Distribution
/aosp_15_r20/external/pytorch/test/distributions/
H A Dtest_distributions.py1324 actual = d.enumerate_support(expand=False)
1327 actual = d.enumerate_support(expand=True)
1375 type(dist.sample()) is type(dist.enumerate_support()),
3986 indep_dist.enumerate_support().shape,
3987 base_dist.enumerate_support().shape,
4926 dist.log_prob(dist.enumerate_support()).size(), torch.Size((3,))
4940 dist.log_prob(dist.enumerate_support()).size(), torch.Size((2, 3))
6240 dist.enumerate_support()
6257 dist.enumerate_support()
6847 return dist.enumerate_support()