Struct tract_rs::Nnef

pub struct Nnef(/* private fields */);

Trait Implementations

impl NnefInterface for Nnef

type Model = Model

fn model_for_path(&self, path: impl AsRef<Path>) -> Result<Model>

Load a NNEF model from the path into a tract-core model.

fn enable_tract_core(&mut self) -> Result<()>

Allow the framework to use tract_core extensions instead of a stricter NNEF definition.

fn enable_tract_extra(&mut self) -> Result<()>

Allow the framework to use tract_extra extensions.

fn enable_onnx(&mut self) -> Result<()>

Allow the framework to use tract_onnx extensions to support operators in ONNX that are absent from NNEF.

fn enable_pulse(&mut self) -> Result<()>

Allow the framework to use tract_pulse extensions to support stateful streaming operation.

fn enable_extended_identifier_syntax(&mut self) -> Result<()>

Allow the framework to use a tract-proprietary extension that can support special characters in node names. If disable, tract will replace everything by underscore ‘_’ to keep compatibility with NNEF. If enabled, the extended syntax will be used, allowing to maintain the node names in serialized form.

fn write_model_to_dir( &self, path: impl AsRef<Path>, model: &Model ) -> Result<()>

Dump a TypedModel as a NNEF directory.

fn write_model_to_tar( &self, path: impl AsRef<Path>, model: &Model ) -> Result<()>

Dump a TypedModel as a NNEF tar file.

fn write_model_to_tar_gz( &self, path: impl AsRef<Path>, model: &Model ) -> Result<()>

fn with_tract_core(self) -> Result<Self, Error>

Convenience function, similar with enable_tract_core but allowing method chaining.

fn with_tract_extra(self) -> Result<Self, Error>

Convenience function, similar with enable_tract_core but allowing method chaining.

fn with_onnx(self) -> Result<Self, Error>

Convenience function, similar with enable_onnx but allowing method chaining.

fn with_pulse(self) -> Result<Self, Error>

Convenience function, similar with enable_pulse but allowing method chaining.

fn with_extended_identifier_syntax(self) -> Result<Self, Error>

Convenience function, similar with enable_extended_identifier_syntax but allowing method chaining.