Struct kn_graph::onnx::GraphLoader

pub struct GraphLoader<'a> { /* private fields */ }

Load an ONNX graph.

Many loading settings are customizable:

• the source, either from a path through Self::from_path or from bytes through Self::from_bytes.
• whether external data is allowed, through Self::from_path allow_external or Self::set_external_data.
• input shape overrides (in order of priority):
  • fully custom through Self::set_input_shaper_custom
  • specific input overrides through Self::force_input_shapes
  • named axes through Self::add_named_axis

A simple example:

// load from a path, disallowing external data
let mut loader = GraphLoader::from_path("model.onnx", false).unwrap();
// set some named axes
loader.add_named_axis("batch_size", Size::BATCH);
loader.add_named_axis("sequence_length", Size::fixed(128));
// override the third input shape
loader.force_input_shapes(vec![None, None, Some(shape![1, Size::BATCH, 3])]);
// load the graph
let graph = loader.load().unwrap();

Implementations

impl<'a> GraphLoader<'a>

pub fn from_path( path: impl AsRef<Path>, allow_external: bool ) -> OnnxResult<Self>

pub fn from_bytes(bytes: &'a [u8]) -> Self

pub fn set_external_data(&mut self, external: Box<dyn ExternalDataLoader>)

pub fn set_input_shaper_custom(&mut self, shaper: Box<InputShaper>)

pub fn force_input_shapes(&mut self, shapes: Vec<Option<Shape>>)

pub fn add_named_axis(&mut self, name: &str, value: Size)

pub fn load(self) -> OnnxResult<Graph>