Struct Model 

pub struct Model { /* private fields */ }

A parsed ONNX model with initializer and edge indexes.

Implementations

impl Model

pub fn load(path: impl AsRef<Path>) -> Result<Self>

Read and parse an ONNX protobuf file.

pub fn from_bytes(bytes: &[u8]) -> Result<Self>

Parse an ONNX protobuf document and build graph indexes.

pub fn name(&self) -> &str

Graph name.

pub fn nodes(&self) -> &[Node]

Nodes in serialized order.

pub fn inputs(&self) -> &[String]

Model input names.

pub fn input_shape(&self, name: &str) -> Result<&[Option<usize>]>

Declared dimensions for a model input. Symbolic dimensions are None.

pub fn opsets(&self) -> &[(String, i64)]

Imported operator-set domains and versions in serialized order.

pub fn outputs(&self) -> &[String]

Model output names.

pub fn initializer(&self, name: &str) -> Result<&Tensor>

Look up an initializer.

pub fn initializers(&self) -> impl Iterator<Item = (&str, &Tensor)>

Initializers in serialized order.

pub fn producer(&self, value: &str) -> Option<&Node>

Node producing a value, when it is not a graph input or initializer.

pub fn consumers<'a>( &'a self, value: &'a str, ) -> impl Iterator<Item = &'a Node> + 'a

Nodes consuming a value.