Struct LayerStack 

pub struct LayerStack { /* private fields */ }

Stack transformer sub-blocks into one named layer stage.

Implementations

impl LayerStack

pub fn new() -> Self

pub fn named(name: impl Into<String>) -> Self

pub fn layer_norm( self, gamma_key: impl Into<String>, beta_key: impl Into<String>, eps: f32, ) -> Self

pub fn gather_add( self, input_name: impl Into<String>, weight_key: impl Into<String>, ) -> Self

pub fn rms_norm(self, weight_key: impl Into<String>, eps: f32) -> Self

pub fn linear(self, weight_key: impl Into<String>, transpose: bool) -> Self

pub fn residual_save(self) -> Self

pub fn residual_add(self) -> Self

pub fn swiglu( self, gate_key: impl Into<String>, up_key: impl Into<String>, down_key: impl Into<String>, ) -> Self

pub fn swiglu_hf_mlp(self, prefix: impl Into<String>) -> Self

pub fn self_attn_prefill(self, spec: SelfAttnPrefillSpec) -> Self

pub fn stage(self, stage: FlowStage) -> Self

pub fn stages(self, stages: impl IntoIterator<Item = FlowStage>) -> Self

pub fn build(self) -> FlowStage

Trait Implementations

impl Clone for LayerStack

fn clone(&self) -> LayerStack

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LayerStack

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

Formats the value using the given formatter.

impl Default for LayerStack

fn default() -> LayerStack

Returns the “default value” for a type.