mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-23 12:12:39 +08:00
ace2b6917b
* Add 0.18.2 Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> * Remove doctree Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> * Add 0.19.0rc0 Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> * Add 0.20.0rc0 Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com> * Add latest 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:
|