burn-onnx 0.21.0-pre.3

Library for importing ONNX models into the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

/// The graph module.
pub mod graph;

mod codegen;
mod node_codegen; // Implements NodeCodegen<PS> for onnx_ir::Node
mod node_traits;

mod imports;

mod argument_helpers;
mod partition;
mod scope;

pub(crate) use argument_helpers::*;
pub(crate) use codegen::ToTokens;
pub(crate) use imports::*;
pub(crate) use node_traits::{Field, TensorKind};
pub(crate) use scope::*;

pub(crate) mod node;