mtia.memory.rst

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