Crate burn_tch

Source
Expand description

Burn Tch Backend

Structs§

LibTorch
Tensor backend that uses LibTorch with the tch crate for executing tensor operations.
TchQTensor
A quantized tensor for the tch backend.
TchShape
A shape that can be used by LibTorch.
TchTensor
A tensor using the tch backend.

Enums§

LibTorchDevice
The device struct when using the tch backend.
Storage
A reference to a tensor storage.

Traits§

QuantElement
A quantized element for the tch backend.
TchElement
The element type for the tch backend.

Functions§

dummy_cuda_dependency
Dummy function to get CUDA to link properly

Type Aliases§

StorageRef
A reference to a tensor storage.