Re-exports§
pub use cpu_runtime::CpuDevice;pub use cpu_runtime::CpuStorage;pub use cpu_runtime::CpuStorageRef;pub use cpu_runtime::CpuStorageRefMut;pub use dtype::DType;pub use dtype::WithDType;pub use error::Error;pub use error::Result;pub use layout::Dim;pub use layout::Layout;pub use layout::Shape;pub use layout::D;pub use lazy_buffer::LazyBuffer;pub use const::Const;pub use schedule::Schedule;pub use schedule::ScheduleItem;
Modules§
- block
- cache
- Compilation cache utilities.
- common_
tests - const
- cpu_
code_ gen - cpu_
runtime - display
- dtype
- error
- interpreter
- lang
- layout
- The shape of a tensor is a tuple with the size of each of its dimensions.
- lazy_
buffer - lower
- lower_
op - safetensors
- samples
- schedule
- utils