torch.accelerator.memory.memory\_stats ====================================== .. currentmodule:: torch.accelerator.memory .. autofunction:: memory_stats