Struct WeightStore 

pub struct WeightStore {
    pub weights: HashMap<ValueId, WeightRef>,
    /* private fields */
}

A resolved set of initializer weights, keyed by the value they populate, plus the live memory maps backing any external data files.

Fields

weights: HashMap<ValueId, WeightRef>

IR weight descriptors, keyed by the graph value they initialize.

Implementations

impl WeightStore

pub fn new() -> Self

An empty store.

pub fn map_external( &mut self, path: impl AsRef<Path>, ) -> Result<(), LoaderError>

Memory-map an external-data file and register it under path, so any WeightRef::External whose path matches resolves zero-copy via bytes. Idempotent: mapping the same path twice is a no-op. This is the programmatic counterpart to the loader path (which maps files while resolving TensorProto initializers), useful when constructing a store by hand.

The map is read-only and kept alive for the store’s lifetime; callers must not mutate or unlink the file while the store is live.

pub fn bytes<'a>(&'a self, weight: &'a WeightRef) -> Option<&'a [u8]>

Resolve a weight descriptor to its raw little-endian bytes.

For inline weights this borrows the stored bytes; for external weights it slices into the memory-mapped file. Returns None if an external mapping is missing or the [offset, offset+length) window is out of bounds.

pub fn external_mmap_provenance( &self, weight: &WeightRef, ) -> Option<(usize, usize, usize)>

Return stable mmap identity and the validated absolute tensor range.

Trait Implementations

impl Debug for WeightStore

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

Formats the value using the given formatter.

impl Default for WeightStore

fn default() -> WeightStore

Returns the “default value” for a type.