mistralrs::layers

Function group_norm

Source
pub fn group_norm(
    num_groups: usize,
    num_channels: usize,
    eps: f64,
    vb: VarBuilderArgs<'_, ShardedSafeTensors<'_>>,
) -> Result<GroupNorm, Error>