.. _torch_tensorrt_py:

torch_tensorrt.runtime
==============================

.. automodule:: torch_tensorrt.runtime
   :members:
   :undoc-members:
   :show-inheritance:

Functions
------------

.. autofunction:: set_multi_device_safe_mode


Classes
---------

.. autoclass:: TorchTensorRTModule
   :members:
   :special-members: __init__

.. autoclass:: PythonTorchTensorRTModule
   :members:
   :special-members: __init__