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 51 52 53 54 55 56 57 58 59 60 61 62
#[macro_use] extern crate derive_new; #[macro_use] extern crate educe; #[macro_use] extern crate log; #[macro_use] pub mod macros; pub mod framework; pub mod infer; pub extern crate tract_core; pub use tract_core::prelude::tract_ndarray; pub use tract_core::prelude::tract_num_traits; pub mod ops { pub mod activations; pub mod array; pub mod binary; pub use tract_core::ops::cast::cast; pub mod cnn; pub mod downsample; pub mod dummy; pub mod element_wise; pub mod expandable; pub mod identity; pub mod konst; pub mod logic; pub use tract_core::ops::math; pub mod matmul; pub mod nn; pub use tract_core::ops::quant; pub mod scan; pub mod source; pub mod unimpl; } pub mod prelude { pub use crate::infer::InferenceFact; pub use crate::infer::InferenceModel; pub use crate::infer::InferenceModelExt; pub use crate::infer::InferenceSimplePlan; pub use tract_core::prelude::*; } pub mod internal { pub use super::prelude::*; pub use crate::infer::*; pub use crate::ops::binary::IntoHir; pub use crate::ops::expandable::{expand, Expansion}; pub use tract_core::internal::*; pub use {shapefactoid, to_typed}; } #[cfg(test)] #[allow(dead_code)] fn setup_test_logger() { let _ = env_logger::Builder::from_env("TRACT_LOG").try_init(); }