Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
layernorm.rst 298 Bytes
Newer Older
Michael Carilli's avatar
Adding layernorm docs  
Michael Carilli committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
.. role:: hidden
    :class: hidden-section

apex.normalization.fused_layer_norm
===================================

.. automodule:: apex.normalization.fused_layer_norm
.. currentmodule:: apex.normalization.fused_layer_norm

.. FusedAdam
   ----------

.. autoclass:: FusedLayerNorm
    :members: