Trait MegaPassPlanView 

pub trait MegaPassPlanView {
    // Required methods
    fn weight_layout(&self) -> MegaPassWeightLayout;
    fn layer_count(&self) -> usize;
    fn layer_weights(&self, layer: usize) -> Option<MegaPassLayerWeights<'_>>;
    fn layer_dims(&self, layer: usize) -> Option<&MegaPassLayerDims>;
    fn layer_weight_keys(&self, layer: usize) -> Option<[u64; 7]>;
    fn layer_norm_keys(&self, layer: usize) -> Option<[u64; 2]>;
}

Immutable view of the cold-prepared weights and tensor metadata.

Required Methods

fn weight_layout(&self) -> MegaPassWeightLayout

fn layer_count(&self) -> usize

fn layer_weights(&self, layer: usize) -> Option<MegaPassLayerWeights<'_>>

fn layer_dims(&self, layer: usize) -> Option<&MegaPassLayerDims>

fn layer_weight_keys(&self, layer: usize) -> Option<[u64; 7]>

fn layer_norm_keys(&self, layer: usize) -> Option<[u64; 2]>

Implementors