Trait finalfusion::storage::Storage

pub trait Storage {
    fn embedding(&self, idx: usize) -> CowArray<f32, Ix1>;
    fn shape(&self) -> (usize, usize);
}

Embedding matrix storage.

To allow for embeddings to be stored in different manners (e.g. regular n x d matrix or as quantized vectors), this trait abstracts over concrete storage types.

Required methods

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

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

Loading content...

Implementors

impl Storage for StorageViewWrap

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

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

impl Storage for StorageWrap

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

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

impl Storage for MmapArray

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

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

impl Storage for MmapQuantizedArray

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

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

impl Storage for NdArray

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

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

impl Storage for QuantizedArray

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

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

Loading content...