Searched defs:empty_cache (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/xpu/ | ||
| D | memory.py | 13 def empty_cache() -> None: function |
| /external/pytorch/torch/mps/ | ||
| D | __init__.py | 81 def empty_cache() -> None: function |
| /external/pytorch/aten/src/ATen/core/ | ||
| D | CachingHostAllocator.h | 212 virtual void empty_cache() { in empty_cache() function |
| /external/pytorch/aten/src/ATen/cuda/ | ||
| D | CachingHostAllocator.cpp | 54 void empty_cache() { in empty_cache() function in at::cuda::__anondde4d04d0111::EventPool |
| /external/pytorch/torch/cuda/ | ||
| D | memory.py | 180 def empty_cache() -> None: function |
| /external/pytorch/test/ | ||
| D | test_cuda_multigpu.py | 149 def assert_change(comp=1, empty_cache=False, reset_peak=False): argument |
| /external/pytorch/c10/cuda/ | ||
| D | CUDACachingAllocator.cpp | 799 void empty_cache() { in empty_cache() function in c10::cuda::CUDACachingAllocator::Native::__anon9cb273fa0111::EventPool |