onnx-ir 0.20.0

ONNX-IR is a pure Rust library for parsing ONNX models into an intermediate representation that can be used to generate code for various ML/DL frameworks
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

//! ONNX Intermediate Representation (IR) types
//!
//! This module contains all the core types for representing ONNX models
//! in an intermediate representation suitable for code generation and analysis.

// Module declarations
mod argument;
mod attribute;
pub(crate) mod graph;
mod node;
mod tensor_data_ext;

pub(crate) use attribute::{AttributeValue, Attributes, DeferredGraph, OuterScopeTypes};
pub(crate) use graph::OnnxGraphBuilder;
pub use node::RuntimeInputRef;
pub(crate) use node::{NodeType, RawNode};
pub use tensor_data_ext::TensorDataExt;

// Re-exports
pub use argument::{ArgType, Argument, DataId, Rank, Shape, TensorType, ValueSource};
pub use burn_tensor::DType;
pub use graph::OnnxGraph;
pub use node::Node;
pub use tensor_data_ext::TensorData;