mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
70 lines
1.2 KiB
ReStructuredText
70 lines
1.2 KiB
ReStructuredText
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:
|