List of all items

Structs

• backprop::GradStore
• dynamic_shape::ShapeEnv
• dynamic_shape::ShapeGuard
• dynamic_shape::SymbolicShape
• layout::Layout
• layout::StridedIter
• op::TensorId
• shape::Shape
• tensor::Tensor

Enums

• backend::BinaryOp
• backend::CmpOp
• backend::ReduceOp
• backend::UnaryOp
• dtype::DType
• dynamic_shape::SymDim
• error::Error
• op::Op

Traits

• backend::Backend
• backend::BackendDevice
• backend::BackendStorage
• dtype::WithDType

Macros

• bail

Functions

• backprop::backward
• backprop::checkpoint
• backprop::checkpoint_sequential
• backprop::is_checkpointing
• backprop::with_checkpoint_mode

Type Aliases

• error::Result