Struct ONNXModel 

pub struct ONNXModel {
    pub ir_version: i64,
    pub producer_name: String,
    pub producer_version: String,
    pub graph: ONNXGraph,
}

ONNX model

Fields

ir_version: i64

producer_name: String

producer_version: String

graph: ONNXGraph

Implementations

impl ONNXModel

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

Create a new ONNX model

pub fn save(&self, path: &str) -> Result<()>

Export model to ONNX file

pub fn load(path: &str) -> Result<Self>

Load ONNX model from file

pub fn add_node(&mut self, node: ONNXNode)

Add a node to the graph

pub fn add_initializer(&mut self, tensor: ONNXTensor)

Add an initializer (weight tensor)

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.