Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributions/
H A Dtransforms.py139 def with_cache(self, cache_size=1): member in Transform
249 def with_cache(self, cache_size=1): member in _InverseTransform
351 def with_cache(self, cache_size=1): member in ComposeTransform
421 def with_cache(self, cache_size=1): member in IndependentTransform
502 def with_cache(self, cache_size=1): member in ReshapeTransform
574 def with_cache(self, cache_size=1): member in PowerTransform
744 def with_cache(self, cache_size=1): member in AffineTransform
1063 def with_cache(self, cache_size=1): member in CatTransform
1155 def with_cache(self, cache_size=1): member in StackTransform
1244 def with_cache(self, cache_size=1): member in CumulativeDistributionTransform