Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
paddle.rst 1.21 KB
Newer Older
Kirthi Shankar Sivamani's avatar
[paddle] add documentation (#489)  
Kirthi Shankar Sivamani committed
1 
..
Kirthi Shankar Sivamani's avatar
Update copyright to include 2025 (#1388)  
Kirthi Shankar Sivamani committed
2 
    Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
Kirthi Shankar Sivamani's avatar
[paddle] add documentation (#489)  
Kirthi Shankar Sivamani committed
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 

    See LICENSE for license information.

paddle
======

.. autoapiclass:: transformer_engine.paddle.Linear(in_features, out_features, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.LayerNorm(hidden_size, eps=1e-5, **kwargs)

.. autoapiclass:: transformer_engine.paddle.LayerNormLinear(in_features, out_features, eps=1e-5, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.LayerNormMLP(hidden_size, ffn_hidden_size, eps=1e-5, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.FusedScaleMaskSoftmax(attn_mask_type, mask_func, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.DotProductAttention(num_attention_heads, kv_channels, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.MultiHeadAttention(hidden_size, num_attention_heads, **kwargs)
  :members: forward

.. autoapiclass:: transformer_engine.paddle.TransformerLayer(hidden_size, ffn_hidden_size, num_attention_heads, **kwargs)
  :members: forward

.. autoapifunction:: transformer_engine.paddle.fp8_autocast

.. autoapifunction:: transformer_engine.paddle.recompute