Enum Mlp

pub enum Mlp {
    NoGating {
        linear1: MaybeQuantizedLinear,
        linear2: MaybeQuantizedLinear,
    },
    Gating {
        linear_in: MaybeQuantizedLinear,
        linear_out: MaybeQuantizedLinear,
        activation: Activation,
    },
}

Variants

NoGating

Fields

linear1: MaybeQuantizedLinear

linear2: MaybeQuantizedLinear

Gating

Fields

linear_in: MaybeQuantizedLinear

linear_out: MaybeQuantizedLinear

activation: Activation

Implementations

impl Mlp

pub fn new(cfg: &Config, vb: MaybeQuantizedVarBuilder<'_>) -> Result<Self>

Trait Implementations

impl Clone for Mlp

fn clone(&self) -> Mlp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Mlp

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

Formats the value using the given formatter.

impl Module for Mlp

fn forward(&self, xs: &Tensor) -> Result<Tensor>