Back to Pytorch

torch.mtia.memory

docs/source/mtia.memory.md

2.11.0330 B
Original Source

torch.mtia.memory

The MTIA backend is implemented out of the tree, only interfaces are defined here.

{eval-rst}
.. automodule:: torch.mtia.memory
{eval-rst}
.. currentmodule:: torch.mtia.memory
{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:

    memory_stats
    memory_allocated