Struct ModelLoader 

pub struct ModelLoader { /* private fields */ }

Loads .oinf model files and exposes tensors/metadata.

Implementations

impl ModelLoader

pub fn open(path: impl AsRef<Path>) -> Result<Self>

Open an .oinf model file from disk.

Example

let model = ModelLoader::open("model.oinf")?;

pub fn size_of(&self, name: &str) -> Result<usize>

Lookup a size variable by name.

Example

let model = ModelLoader::open("model.oinf")?;
let b = model.size_of("B")?;

pub fn resolve_len(&self, dims: &[String]) -> Result<usize>

Resolve a product of dimension strings into a length.

pub fn resolve_shape(&self, dims: &[String]) -> Result<Vec<usize>>

Resolve dimension strings into a concrete shape.

Example

let model = ModelLoader::open("model.oinf")?;
let shape = model.resolve_shape(&["B".into(), "D".into()])?;

pub fn resolve_dim_value(&self, dim: &str) -> Result<usize>

Resolve a single dimension expression (literal, sizevar, or product).

pub fn var_info(&self, name: &str) -> Option<&VarInfo>

Fetch variable metadata by name.

pub fn tensor_store(&self) -> &TensorStore

Access the underlying tensor store.

pub fn load_tensor(&self, name: &str) -> Result<TensorValue>

Load a tensor payload by name from the mapped file.

Example

let model = ModelLoader::open("model.oinf")?;
let tensor = model.load_tensor("w1")?;

pub fn load_metadata_tensor(&self, name: &str) -> Result<Option<TensorValue>>

Load a metadata tensor by name, if present.

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

True if a named metadata entry is a string.

pub fn load_metadata_string(&self, name: &str) -> Result<Option<String>>

Load a metadata string by name, if present.

Trait Implementations

impl Clone for ModelLoader

fn clone(&self) -> ModelLoader

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelLoader

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.