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 90e3644d..2a36cebd 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -65,6 +65,7 @@ pub mod scalar; pub mod shape; mod sort; mod storage; +pub mod streaming; mod strided_index; mod tensor; mod tensor_cat; @@ -84,6 +85,7 @@ pub use indexer::IndexOp; pub use layout::Layout; pub use shape::{Shape, D}; pub use storage::Storage; +pub use streaming::{StreamTensor, StreamingBinOp, StreamingModule}; pub use strided_index::{StridedBlocks, StridedIndex}; pub use tensor::{Tensor, TensorId}; pub use variable::Var; |