Crate torch_sys2

Crate torch_sys2 

Source
Expand description

Simplified Rust bindings for libtorch C++ APIs.

This is a streamlined version that only includes the functionality actually used by the monarch codebase.

Modules§

testing
Testing utilities

Structs§

Borrow
BorrowGuard
BorrowGuardMut
CudaDevice
Device
DeviceIndex
MultiBorrow
Tensor
TensorCell

Enums§

BorrowError
BorrowType
DeviceParseError
DeviceType
Layout
LayoutDef
Remote serde implementation for Layout
MemoryFormat
MemoryFormatDef
Remote serde implementation for MemoryFormat
ScalarType
ScalarTypeDef
Remote serde implementation for ScalarType

Traits§

CloneUnsafe

Functions§

deep_clone
factory_empty
factory_float_tensor
factory_zeros
is_float8_type
suggest_memory_format