Struct stablediffusion_wgpu::model::unet::TransformerBlockRecord

pub struct TransformerBlockRecord<B: Backend> {
    pub norm1: <LayerNorm<B> as Module<B>>::Record,
    pub attn1: <MultiHeadAttention<B> as Module<B>>::Record,
    pub norm2: <LayerNorm<B> as Module<B>>::Record,
    pub attn2: <MultiHeadAttention<B> as Module<B>>::Record,
    pub norm3: <LayerNorm<B> as Module<B>>::Record,
    pub mlp: <MLP<B> as Module<B>>::Record,
}

The record type for the module.

Fields

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

The module record associative type.

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

The module record associative type.

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

The module record associative type.

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

The module record associative type.

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

The module record associative type.

mlp: <MLP<B> as Module<B>>::Record

The module record associative type.

Trait Implementations

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

fn clone(&self) -> TransformerBlockRecord<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 TransformerBlockRecord<B>

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

Formats the value using the given formatter.

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

type Item<S: PrecisionSettings> = TransformerBlockRecordItem<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.