huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

PyTorch like display implementation.

Browse Source
This commit is contained in:
laurent
2023-06-27 21:16:35 +01:00
parent 934655a60d
commit 8c81a70170
3 changed files with 265 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/lib.rs
View File

@ -3,7 +3,7 @@ mod cpu_backend;
#[cfg(feature = "cuda")]
mod cuda_backend;
mod device;
mod display;
pub mod display;
mod dtype;
mod dummy_cuda_backend;
mod error;
@ -13,7 +13,7 @@ mod shape;
mod storage;
mod strided_index;
mod tensor;
mod utils;
pub mod utils;
pub use cpu_backend::CpuStorage;
pub use device::{Device, DeviceLocation};