Home
last modified time | relevance | path

Searched refs:_cache_enabled (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/nn/utils/
Dparametrize.py27 _cache_enabled = 0 variable
63 global _cache_enabled
64 _cache_enabled += 1
68 _cache_enabled -= 1
69 if not _cache_enabled:
391 if _cache_enabled:
/external/pytorch/torch/amp/
Dautocast_mode.py260 self._cache_enabled = torch.is_autocast_cache_enabled()
273 self._cache_enabled = cache_enabled
356 torch.set_autocast_cache_enabled(self._cache_enabled)
/external/pytorch/torch/_dynamo/variables/
Dbuilder.py976 value._cache_enabled,