Skip to main content

Module rms_norm_gated

burn_mamba::utils

Module rms_norm_gated 

Source

Structsยง

RmsNormGated
Applies Gated Rms Normalization over an input tensor along the last dimension.
RmsNormGatedConfig
Configuration to create a RmsNormGated layer.
RmsNormGatedRecord
The record type for the module.
RmsNormGatedRecordItem
The record item type for the module.