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
#[allow(unused_imports)] #[macro_use] extern crate derive_new; #[macro_use] extern crate error_chain; #[allow(unused_imports)] #[macro_use] extern crate log; extern crate num_integer; extern crate num_traits; #[allow(unused_imports)] #[macro_use] extern crate tract_core; extern crate tract_linalg; pub mod model; pub mod ops; pub mod pb { include!(concat!(env!("OUT_DIR"), "/prost/onnx.rs")); } pub mod pb_helpers; pub mod tensor; pub use model::Onnx; use tract_core::internal::*; #[deprecated(note = "Please use onnx().model_for_path(..)")] pub fn for_path(p: impl AsRef<std::path::Path>) -> TractResult<InferenceModel> { onnx().model_for_path(p) } #[deprecated(note = "Please use onnx().model_for_read(..)")] pub fn for_reader<R: std::io::Read>(mut r: R) -> TractResult<InferenceModel> { onnx().model_for_read(&mut r) } pub fn onnx() -> Onnx { let mut ops = crate::model::OnnxOpRegister::default(); ops::register_all_ops(&mut ops); Onnx { op_register: ops } }