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

mod num;
pub use num::*;

mod buffer;
pub use buffer::*;

mod tensor;
pub use tensor::*;

pub mod prelude {
    pub use crate::{TensorTrait as _};
}

/// Host-only profile.
/// Import all from it if you are planning to use only host-side tensors.
pub mod host_only {
    pub use super::{
        *,
        HostBuffer as Buffer,
        HostTensor as Tensor,
    };
}

#[cfg(feature = "device")]
/// Host and device profile.
/// Import all from it if you are planning to use both host-side and device-side tensors.
pub mod host_device {
    pub use super::{
        *,
        TensorTrait as Tensor,
    };
}