pytorch/docs/source/mtia.memory.rst

14 lines
292 B
ReStructuredText

torch.mtia.memory
===================================
The MTIA backend is implemented out of the tree, only interfaces are be defined here.
.. automodule:: torch.mtia.memory
.. currentmodule:: torch.mtia.memory
.. autosummary::
:toctree: generated
:nosignatures:
memory_stats