mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
85a2a656c2
* Update switcher.json Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> * Update 0.19 doc Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> --------- Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com>
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:
|