kn-graph 0.7.4

A neural network inference graph intermediate representation, with surrounding utilities.
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
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use std::path::Path;

use external_data::ExternalDataLoader;
pub use load::{InputShaper, OnnxDimValue};
pub use loader::GraphLoader;

use crate::graph::Graph;
use crate::onnx::result::OnnxResult;
use crate::shape::Size;

pub mod external_data;
mod inputs;
mod load;
#[allow(warnings)]
mod proto;
pub mod result;
mod store;
mod typed_value;
mod loader;

/// Load an [ONNX](https://github.com/onnx/onnx/blob/main/docs/IR.md) file from the given path.
///
/// If `allow_external` is true, the onnx will be allowed to load external data files,
/// see [the spec](https://github.com/onnx/onnx/blob/main/docs/IR.md#external-tensor-data).
/// If `allow_external` is false and the ONNX file does reference external data, an error is returned.
///
/// For more flexibility, see [GraphLoader].
pub fn load_graph_from_onnx_path(path: impl AsRef<Path>, allow_external: bool) -> OnnxResult<Graph> {
    let mut loader = GraphLoader::from_path(path, allow_external)?;
    loader.add_named_axis("batch_size", Size::BATCH);
    loader.load()
}

/// Load an [ONNX](https://github.com/onnx/onnx/blob/main/docs/IR.md) file from the given bytes.
///
/// The file is not allowed to reference external data files.
///
/// For more flexibility, see [GraphLoader].
pub fn load_graph_from_onnx_bytes(buffer: &[u8]) -> OnnxResult<Graph> {
    let mut loader = GraphLoader::from_bytes(buffer);
    loader.add_named_axis("batch_size", Size::BATCH);
    loader.load()
}

pub fn load_graph_from_onnx_bytes_custom(buffer: &[u8], external: Box<dyn ExternalDataLoader>) -> OnnxResult<Graph> {
    let mut loader = GraphLoader::from_bytes(buffer);
    loader.set_external_data(external);
    loader.add_named_axis("batch_size", Size::BATCH);
    loader.load()
}