Blame · docs/source/layernorm.rst · a786ca0cf2d2dd790cecd96b19cc478d7e661cb5 · OpenDAS / apex · GitLab

Find file Normal view History Permalink

layernorm.rst 307 Bytes

Newer Older

Adding layernorm docs Michael Carilli committed Nov 28, 2018	1 2 3 4 5 6	`.. role:: hidden :class: hidden-section apex.normalization.fused_layer_norm ===================================`
Shortening import path for layernorm Michael Carilli committed Nov 28, 2018	7 8	`.. automodule:: apex.normalization .. currentmodule:: apex.normalization`
Adding layernorm docs Michael Carilli committed Nov 28, 2018	9 10 11 12 13 14	`.. FusedAdam ---------- .. autoclass:: FusedLayerNorm :members:`
fix and generate docs for FusedRMSNorm (#1285) eqy committed Feb 07, 2022	15 16 17	`.. autoclass:: FusedRMSNorm :members:`