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
63
#[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, inference_wrap, Expansion};
pub use tract_core;
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();
}