.. _torch_tensorrt_fx_py:

torch_tensorrt.fx
===================

.. currentmodule:: torch_tensorrt.fx

.. automodule torch_tensorrt.ts
   :undoc-members:

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

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

.. autofunction:: compile


Classes
--------

.. autoclass:: TRTModule

.. autoclass:: InputTensorSpec

.. autoclass:: TRTInterpreter

.. autoclass:: TRTInterpreterResult