List of all items
Structs
- Device
- gradtrack::NoGradTrack
- optimizers::Adam
- optimizers::AdamConfig
- serialization::StructDeserializer
- serialization::StructSerializer
- tensor::NoMemPoolGuard
- tensor::Tensor
Enums
- DeviceType
- serialization::FieldValue
- serialization::Format
- serialization::SerializationError
- tensor::Shape
Traits
- optimizers::Optimizer
- serialization::FromFieldValue
- serialization::Serializable
- serialization::StructSerializable
- serialization::ToFieldValue
- tensor::TensorCollectExt
- tensor::ValuesCollectExt
Functions
- cuda_device_count
- cuda_is_available
- current_device
- get_default_device
- gradtrack::clear_all_graphs_known
- gradtrack::clear_all_shared_graphs
- gradtrack::clear_gradient_for_tensor
- gradtrack::clear_gradients
- gradtrack::clear_graph_for_tensor
- gradtrack::clear_local_graph
- gradtrack::clear_shared_graph_for_tensor
- gradtrack::is_grad_enabled
- gradtrack::set_grad_enabled
- gradtrack::with_no_grad
- set_default_device
- tensor::with_no_mem_pool
- with_device