Module device 

Device abstraction for CPU, CUDA, and Metal backends.

use fuel_tensor::Device;
let dev = Device::cpu();
assert!(dev.is_cpu());
assert_eq!(dev.location(), fuel_tensor::DeviceLocation::Cpu);

Structs

Device

A device on which tensors can be created and computations performed.

Enums

DeviceLocation

Identifies a physical device location (CPU or a specific GPU).

Traits

NdArray

Trait for host array-like values that describe their shape and yield CPU storage.