Searched defs:with_cache (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/torch/distributions/ |
H A D | transforms.py | 139 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
|