InferenceModel

Type Alias InferenceModel 

Source
pub type InferenceModel = Graph<InferenceFact, Box<dyn InferenceOp>>;
Expand description

A model with partially types and shapes, as produced by parsing ONNX or Tensorflow graphs.

Aliased Type§

pub struct InferenceModel {
    pub nodes: Vec<Node<InferenceFact, Box<dyn InferenceOp>>>,
    pub inputs: Vec<OutletId>,
    pub outputs: Vec<OutletId>,
    pub outlet_labels: HashMap<OutletId, String>,
    pub properties: HashMap<String, Arc<Tensor>>,
    pub symbols: SymbolScope,
}

Fields§

§nodes: Vec<Node<InferenceFact, Box<dyn InferenceOp>>>

all nodes in the model

§inputs: Vec<OutletId>

model inputs

§outputs: Vec<OutletId>

model outputs

§outlet_labels: HashMap<OutletId, String>

outlet labels

§properties: HashMap<String, Arc<Tensor>>

model properties

§symbols: SymbolScope

symbol scope, including table