Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dindependent.py79 def has_enumerate_support(self): member in Independent
82 return self.base_dist.has_enumerate_support
H A Dbernoulli.py41 has_enumerate_support = True variable in Bernoulli
H A Done_hot_categorical.py43 has_enumerate_support = True variable in OneHotCategorical
H A Dcategorical.py51 has_enumerate_support = True variable in Categorical
H A Dbinomial.py49 has_enumerate_support = True variable in Binomial
H A Ddistribution.py21 has_enumerate_support = False variable in Distribution