Enum finalfusion::storage::StorageWrap

pub enum StorageWrap {
    NdArray(NdArray),
    QuantizedArray(Box<QuantizedArray>),
    MmapArray(MmapArray),
    MmapQuantizedArray(MmapQuantizedArray),
}

Storage types wrapper.

This crate makes it possible to create fine-grained embedding types, such as Embeddings<SimpleVocab, NdArray> or Embeddings<SubwordVocab, QuantizedArray>. However, in some cases it is more pleasant to have a single type that covers all vocabulary and storage types. VocabWrap and StorageWrap wrap all the vocabularies and storage types known to this crate such that the type Embeddings<VocabWrap, StorageWrap> covers all variations.

Variants

NdArray(NdArray)

QuantizedArray(Box<QuantizedArray>)

MmapArray(MmapArray)

MmapQuantizedArray(MmapQuantizedArray)

Trait Implementations

impl From<MmapArray> for StorageWrap

fn from(s: MmapArray) -> Self

Performs the conversion.

impl From<MmapQuantizedArray> for StorageWrap

fn from(s: MmapQuantizedArray) -> Self

Performs the conversion.

impl From<NdArray> for StorageWrap

fn from(s: NdArray) -> Self

Performs the conversion.

impl From<QuantizedArray> for StorageWrap

fn from(s: QuantizedArray) -> Self

Performs the conversion.

impl Storage for StorageWrap

fn embedding(&self, idx: usize) -> CowArray<f32, Ix1>

fn shape(&self) -> (usize, usize)