sglang_v0.5.2/flashinfer_0.3.1/docs/api/norm.rst

17 lines
243 B
ReStructuredText

.. _apinorm:
flashinfer.norm
===============
Kernels for normalization layers.
.. currentmodule:: flashinfer.norm
.. autosummary::
:toctree: ../generated
rmsnorm
fused_add_rmsnorm
gemma_rmsnorm
gemma_fused_add_rmsnorm