Trait SimpleBackend

pub trait SimpleBackend: Send + Sync {
    // Required methods
    fn get(
        &self,
        s: Shape,
        name: &str,
        h: Init,
        dtype: DType,
        dev: &Device,
    ) -> Result<Tensor>;
    fn get_unchecked(
        &self,
        name: &str,
        dtype: DType,
        dev: &Device,
    ) -> Result<Tensor>;
    fn contains_tensor(&self, name: &str) -> bool;
}

Required Methods

fn get( &self, s: Shape, name: &str, h: Init, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor based on a target name and shape.

fn get_unchecked( &self, name: &str, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor based on the name.

fn contains_tensor(&self, name: &str) -> bool

Trait Implementations

impl Backend for Box<dyn SimpleBackend + '_>

type Hints = Init

fn get( &self, s: Shape, name: &str, h: Self::Hints, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor with some target shape.

fn get_unchecked( &self, name: &str, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor based on the name.

fn contains_tensor(&self, name: &str) -> bool

Implementations on Foreign Types

impl SimpleBackend for HashMap<String, Tensor>

fn get( &self, s: Shape, name: &str, _: Init, dtype: DType, dev: &Device, ) -> Result<Tensor>

fn get_unchecked( &self, name: &str, dtype: DType, dev: &Device, ) -> Result<Tensor>

fn contains_tensor(&self, name: &str) -> bool

Implementors

impl SimpleBackend for NpzTensors

impl SimpleBackend for PthTensors

impl SimpleBackend for BufferedSafetensors

impl SimpleBackend for MmapedSafetensors

impl SimpleBackend for SliceSafetensors<'_>

impl SimpleBackend for VarMap

impl SimpleBackend for SafeTensorWithRouting<'_>

impl<R: Renamer + Sync + Send> SimpleBackend for Rename<'_, R>