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