Crate tenferro_device

Crate tenferro_device 

Source
Expand description

Device abstraction and shared error types for the tenferro workspace.

This crate provides:

§Examples

use tenferro_device::{LogicalMemorySpace, ComputeDevice};

let space = LogicalMemorySpace::MainMemory;
let dev = ComputeDevice::Cpu { device_id: 0 };
assert_eq!(format!("{dev}"), "cpu:0");

Enums§

ComputeDevice
Compute device that can execute tensor operations.
Error
Error type used across the tenferro workspace.
LogicalMemorySpace
Logical memory space where tensor data resides.
OpKind
Classification of tensor operations, used to query preferred compute devices for a given operation on a given memory space.

Functions§

preferred_compute_devices
Return the preferred compute devices for a given operation on a memory space.

Type Aliases§

Result
Result type alias using Error.