Struct Onnx 

pub struct Onnx {
    pub op_register: OnnxOpRegister,
    pub use_output_shapes: bool,
    pub ignore_output_types: bool,
    pub provider: Arc<dyn ModelDataResolver + Send + Sync>,
}

Fields

op_register: OnnxOpRegister

use_output_shapes: bool

ignore_output_types: bool

provider: Arc<dyn ModelDataResolver + Send + Sync>

Implementations

impl Onnx

pub fn parse( &self, proto: &ModelProto, path: Option<&str>, ) -> TractResult<ParseResult>

pub fn parse_with_template( &self, proto: &ModelProto, model_dir: Option<&str>, template: InferenceModel, ) -> TractResult<ParseResult>

pub fn with_ignore_output_shapes(self, ignore: bool) -> Onnx

pub fn with_ignore_output_types(self, ignore: bool) -> Onnx

pub fn determinize(model: &mut InferenceModel) -> TractResult<()>

Trait Implementations

impl Clone for Onnx

fn clone(&self) -> Onnx

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for Onnx

fn default() -> Self

Returns the “default value” for a type.

impl Framework<ModelProto, Graph<InferenceFact, Box<dyn InferenceOp>>> for Onnx

fn model_for_path(&self, p: impl AsRef<Path>) -> TractResult<InferenceModel>

Build a model from a filename.

fn proto_model_for_path(&self, p: impl AsRef<Path>) -> TractResult<ModelProto>

Read a proto model from a filename.

fn proto_model_for_read(&self, r: &mut dyn Read) -> TractResult<ModelProto>

Parse a proto model from a reader.

fn model_for_proto_model_with_model_template( &self, proto: &ModelProto, template: InferenceModel, ) -> TractResult<InferenceModel>

Translate a proto model into a model, with some symbols already listed.

fn model_for_read(&self, r: &mut dyn Read) -> TractResult<InferenceModel>

Read a model from a reader

fn model_for_proto_model(&self, proto: &ProtoModel) -> Result<Model, Error>

Translate a proto model into a model.