diff options
Diffstat (limited to 'src/lib.rs')
-rw-r--r-- | src/lib.rs | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -1,3 +1,4 @@ +mod cpu_backend; mod device; mod dtype; mod error; @@ -7,10 +8,11 @@ mod storage; mod strided_index; mod tensor; +pub use cpu_backend::CpuStorage; pub use device::Device; pub use dtype::{DType, WithDType}; pub use error::{Error, Result}; pub use shape::Shape; -pub use storage::{CpuStorage, Storage}; +pub use storage::Storage; use strided_index::StridedIndex; pub use tensor::{Tensor, TensorId}; |