Trait candle_nn::var_builder::SimpleBackend

pub trait SimpleBackend: Send + Sync {
    // Required methods
    fn get(
        &self,
        s: Shape,
        name: &str,
        h: Init,
        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 contains_tensor(&self, name: &str) -> bool

Trait Implementations

impl<'a> Backend for Box<dyn SimpleBackend + 'a>

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 contains_tensor(&self, name: &str) -> bool

Implementations on Foreign Types

impl SimpleBackend for NpzTensors

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

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

impl SimpleBackend for PthTensors

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

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

impl SimpleBackend for BufferedSafetensors

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

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

impl SimpleBackend for MmapedSafetensors

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

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

impl SimpleBackend for HashMap<String, Tensor>

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

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

Implementors

impl SimpleBackend for VarMap

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