Struct burn_core::nn::transformer::TransformerDecoderLayerRecord

pub struct TransformerDecoderLayerRecord<B: Backend> {
    pub cross_attn: <MultiHeadAttention<B> as Module<B>>::Record,
    pub self_attn: <MultiHeadAttention<B> as Module<B>>::Record,
    pub pwff: <PositionWiseFeedForward<B> as Module<B>>::Record,
    pub norm_1: <LayerNorm<B> as Module<B>>::Record,
    pub norm_2: <LayerNorm<B> as Module<B>>::Record,
    pub norm_3: <LayerNorm<B> as Module<B>>::Record,
    pub dropout: <Dropout as Module<B>>::Record,
    pub norm_first: <bool as Module<B>>::Record,
}

The record type for the module.

Fields

cross_attn: <MultiHeadAttention<B> as Module<B>>::Record

The module record associative type.

self_attn: <MultiHeadAttention<B> as Module<B>>::Record

The module record associative type.

pwff: <PositionWiseFeedForward<B> as Module<B>>::Record

The module record associative type.

norm_1: <LayerNorm<B> as Module<B>>::Record

The module record associative type.

norm_2: <LayerNorm<B> as Module<B>>::Record

The module record associative type.

norm_3: <LayerNorm<B> as Module<B>>::Record

The module record associative type.

dropout: <Dropout as Module<B>>::Record

The module record associative type.

norm_first: <bool as Module<B>>::Record

The module record associative type.

Trait Implementations

impl<B: Clone + Backend> Clone for TransformerDecoderLayerRecord<B>

fn clone(&self) -> TransformerDecoderLayerRecord<B>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<B: Debug + Backend> Debug for TransformerDecoderLayerRecord<B>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<B: Backend> Record for TransformerDecoderLayerRecord<B>

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

Type of the item that can be serialized and deserialized.

fn into_item<S: PrecisionSettings>(self) -> Self::Item<S>

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

fn from_item<S: PrecisionSettings>(item: Self::Item<S>) -> Self

Convert the given item into a record.