Struct OnnxModel 

pub struct OnnxModel {
    pub producer_name: String,
    pub producer_version: String,
    pub graph_name: String,
    pub nodes: Vec<OnnxNode>,
    pub initializers: Vec<OnnxTensor>,
    pub inputs: Vec<(String, Vec<i64>, i32)>,
    pub outputs: Vec<(String, Vec<i64>, i32)>,
}

An ONNX model with graph, metadata, and opset information.

Fields

producer_name: String

Model producer name.

producer_version: String

Model producer version.

graph_name: String

Graph name.

nodes: Vec<OnnxNode>

Graph nodes (operations).

initializers: Vec<OnnxTensor>

Initializer tensors (weights).

inputs: Vec<(String, Vec<i64>, i32)>

Graph inputs (names and shapes).

outputs: Vec<(String, Vec<i64>, i32)>

Graph outputs (names and shapes).

Implementations

impl OnnxModel

pub fn new(graph_name: &str) -> Self

Create a new empty ONNX model.

pub fn to_bytes(&self) -> Vec<u8>

Encode to ONNX protobuf binary format.

pub fn save<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save ONNX model to a file.

Trait Implementations

impl Clone for OnnxModel

fn clone(&self) -> OnnxModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OnnxModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.