mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-25 05:02:59 +08:00
5e2cf02f46
update docs for 0.20.0rc2 Signed-off-by: Shixiaowei02 <39303645+Shixiaowei02@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:
|