Type Alias tract_tensorflow::prelude::InferenceModel

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

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

Aliased Type

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 symbol_table: SymbolTable,
}

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

symbol_table: SymbolTable

symbol table