Home
last modified time | relevance | path

Searched defs:enumerate_support (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dindependent.py117 def enumerate_support(self, expand=True): member in Independent
H A Dcategorical.py151 def enumerate_support(self, expand=True): member in Categorical
H A Dbinomial.py155 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 Ddistribution.py218 def enumerate_support(self, expand: bool = True) -> torch.Tensor: member in Distribution