1 2 3 4 5 6 7 8 9 10 11 12
#[cfg(feature = "tch")] use ::tch::Device; pub mod autodiff; pub mod conversion; #[cfg(feature = "ndarray")] pub mod ndarray; #[cfg(feature = "tch")] pub mod tch; #[cfg(feature = "tch")] pub type TchDevice = Device;