Struct SanitizedBackend 

pub struct SanitizedBackend { /* private fields */ }

A SimpleBackend that renames on lookup.

Rather than sanitizing every key up front, the inverse mapping is computed once at load time and consulted per parameter, so the underlying mmap is never copied.

Implementations

impl SanitizedBackend

pub fn new( inner: Box<dyn SimpleBackend>, stored_names: impl IntoIterator<Item = String>, ) -> Self

Trait Implementations

impl SimpleBackend for SanitizedBackend

fn get( &self, shape: Shape, name: &str, init: Init, dtype: DType, device: &Device, ) -> Result<Tensor>

Retrieve a tensor based on a target name and shape.

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

Retrieve a tensor based on the name.

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