Searched refs:max_memory_cached (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/docs/source/ | ||
| D | cuda.rst | 115 max_memory_cached |
| /external/pytorch/torch/cuda/ | ||
| D | memory.py | 486 def max_memory_cached(device: Union[Device, int] = None) -> int: function |