Layers
===========================

.. automodule:: tensorrt_llm

.. currentmodule:: tensorrt_llm

Activation
------------
.. automodule:: tensorrt_llm.layers.activation
   :members:
   :undoc-members:
   :show-inheritance:

Attention
------------
.. automodule:: tensorrt_llm.layers.attention
   :members:
   :undoc-members:
   :show-inheritance:

Cast
------------
.. automodule:: tensorrt_llm.layers.cast
   :members:
   :undoc-members:
   :show-inheritance:

Conv
------------
.. automodule:: tensorrt_llm.layers.conv
   :members:
   :undoc-members:
   :show-inheritance:

Embedding
------------
.. automodule:: tensorrt_llm.layers.embedding
   :members:
   :undoc-members:
   :show-inheritance:

Linear
------------
.. automodule:: tensorrt_llm.layers.linear
   :members:
   :undoc-members:
   :show-inheritance:

MLP
------------
.. automodule:: tensorrt_llm.layers.mlp
   :members:
   :undoc-members:
   :show-inheritance:

Normalization
---------------
.. automodule:: tensorrt_llm.layers.normalization
   :members:
   :undoc-members:
   :show-inheritance:

Pooling
------------
.. automodule:: tensorrt_llm.layers.pooling
   :members:
   :undoc-members:
   :show-inheritance:
