Skip to main content

layer_kinds

Function layer_kinds 

Source
pub fn layer_kinds(
    spec: &ModelSpec,
    source_of: &dyn Fn(usize) -> (String, BTreeSet<String>),
) -> Vec<(String, Vec<usize>)>
Expand description

The model’s layer kinds, in the order their first layer appears: each kind’s name and its layers. Two layers are of one kind when their authored programs (source_of: the text, and the weights it reads as stored Q8_0) are one program up to the names they give — the same shape_digest — so one lowered program serves both, bound to each layer’s own tensors by the renaming between their programs.