Enum MaybeQuantizedEmbedding

pub enum MaybeQuantizedEmbedding {
    Real(Embedding),
    Quantized(Embedding),
}

Variants

Real(Embedding)

Quantized(Embedding)

Implementations

impl MaybeQuantizedEmbedding

pub fn new( in_vocab_size: usize, dim: usize, vb: MaybeQuantizedVarBuilder<'_>, ) -> Result<Self>

pub fn embeddings(&self) -> &Tensor

pub fn hidden_size(&self) -> Result<usize>

pub fn dtype(&self) -> DType

Trait Implementations

impl Clone for MaybeQuantizedEmbedding

fn clone(&self) -> MaybeQuantizedEmbedding

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MaybeQuantizedEmbedding

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

Formats the value using the given formatter.

impl Module for MaybeQuantizedEmbedding

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