diff options
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 0108c198..af1eb215 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -56,6 +56,7 @@ mod storage; mod strided_index; mod tensor; pub mod utils; +mod variable; pub use cpu_backend::CpuStorage; pub use device::{Device, DeviceLocation}; @@ -67,6 +68,7 @@ pub use shape::{Shape, D}; pub use storage::Storage; use strided_index::StridedIndex; pub use tensor::{Tensor, TensorId}; +pub use variable::Variable; #[cfg(feature = "cuda")] pub use cuda_backend::{CudaDevice, CudaStorage}; |