diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-06-28 15:59:53 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-28 15:59:53 +0100 |
| commit | 0cfa21f26a88820fba91bb8ff02cf850eeeb97c3 (patch) | |
| tree | efc1279fc9ba273425689e79ac5577801b1bddae /candle-core/src/lib.rs | |
| parent | 8b4b2d1830e6fb5aed2c410256bb4e7076e5007d (diff) | |
| parent | 6c9e6b5a99d4070be5c20d7c383e0ef7e3228260 (diff) | |
| download | candle-0cfa21f26a88820fba91bb8ff02cf850eeeb97c3.tar.gz candle-0cfa21f26a88820fba91bb8ff02cf850eeeb97c3.tar.bz2 candle-0cfa21f26a88820fba91bb8ff02cf850eeeb97c3.zip | |
Merge pull request #27 from LaurentMazare/layout-refactor
Refactor the stride/shape handling
Diffstat (limited to 'candle-core/src/lib.rs')
| -rw-r--r-- | candle-core/src/lib.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs index 5771517f..6a860116 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -7,6 +7,7 @@ pub mod display; mod dtype; mod dummy_cuda_backend; mod error; +mod layout; mod npy; mod op; mod shape; @@ -19,6 +20,7 @@ pub use cpu_backend::CpuStorage; pub use device::{Device, DeviceLocation}; pub use dtype::{DType, WithDType}; pub use error::{Error, Result}; +pub use layout::Layout; pub use shape::Shape; pub use storage::Storage; use strided_index::StridedIndex; |