autodiff/
lib.rs

1#[global_allocator]
2static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
3
4pub mod adops;
5pub mod autodiff;
6pub mod diffable;
7pub mod compose;
8pub mod autodiffable;
9pub mod autotuple;
10pub mod func_traits;
11pub mod funcs;
12pub mod traits;
13pub mod forward;
14pub mod gradienttype;
15
16// re-export
17pub use autodiff::*;
18pub use diffable::*;
19pub use compose::*;
20pub use autodiffable::*;
21pub use autotuple::*;
22pub use func_traits::*;
23//pub use funcs::*;
24pub use traits::*;
25pub use forward::*;
26pub use gradienttype::*;
27
28#[cfg(feature = "ndarray")]
29pub mod ad_ndarray;
30
31#[cfg(feature = "ndarray")]
32pub use ad_ndarray::*;
33
34#[cfg(test)]
35mod test_autodiff;
36#[cfg(test)]
37mod test_manualdiff;
38
39// re-export derive proc-macros
40pub use autodiff_derive::*;