Enum NormalizationRecord

pub enum NormalizationRecord<B>
where
    B: Backend,
{
    Batch(<BatchNorm<B> as Module<B>>::Record),
    Group(<GroupNorm<B> as Module<B>>::Record),
    Instance(<InstanceNorm<B> as Module<B>>::Record),
    Layer(<LayerNorm<B> as Module<B>>::Record),
    Rms(<RmsNorm<B> as Module<B>>::Record),
}

The record type for the module.

Variants

Batch(<BatchNorm<B> as Module<B>>::Record)

The module record associative type.

Group(<GroupNorm<B> as Module<B>>::Record)

The module record associative type.

Instance(<InstanceNorm<B> as Module<B>>::Record)

The module record associative type.

Layer(<LayerNorm<B> as Module<B>>::Record)

The module record associative type.

Rms(<RmsNorm<B> as Module<B>>::Record)

The module record associative type.

Trait Implementations

impl<B> Record<B> for NormalizationRecord<B>

where B: Backend,

type Item<S: PrecisionSettings> = NormalizationRecordItem<B, S>

Type of the item that can be serialized and deserialized.

fn into_item<S>(self) -> <NormalizationRecord<B> as Record<B>>::Item<S>

where S: PrecisionSettings,

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S>( item: <NormalizationRecord<B> as Record<B>>::Item<S>, device: &<B as Backend>::Device, ) -> NormalizationRecord<B>

where S: PrecisionSettings,

Convert the given item into a record.