burn_ndarray/
lib.rs

1#![cfg_attr(not(feature = "std"), no_std)]
2#![warn(missing_docs)]
3#![cfg_attr(docsrs, feature(doc_auto_cfg))]
4
5//! Burn ndarray backend.
6
7#[macro_use]
8extern crate derive_new;
9
10#[cfg(any(
11    feature = "blas-netlib",
12    feature = "blas-openblas",
13    feature = "blas-openblas-system",
14))]
15extern crate blas_src;
16
17mod backend;
18mod element;
19mod ops;
20mod sharing;
21mod tensor;
22
23pub use backend::*;
24pub use element::FloatNdArrayElement;
25pub(crate) use sharing::*;
26pub use tensor::*;
27
28extern crate alloc;
29
30#[cfg(test)]
31mod tests {
32    type TestBackend = crate::NdArray<f32>;
33    type TestTensor<const D: usize> = burn_tensor::Tensor<TestBackend, D>;
34    type TestTensorInt<const D: usize> = burn_tensor::Tensor<TestBackend, D, burn_tensor::Int>;
35    type TestTensorBool<const D: usize> = burn_tensor::Tensor<TestBackend, D, burn_tensor::Bool>;
36
37    use alloc::format;
38    use alloc::vec;
39
40    burn_tensor::testgen_all!();
41    burn_tensor::testgen_quantization!();
42
43    #[cfg(feature = "std")]
44    burn_autodiff::testgen_all!();
45}
burn_ndarray/lib.rs

burn_ndarray/
lib.rs
