nablatensor

Docs / nablatensor-tensor / com.nablatensor.tensor

record

Device

A concrete compute device: a DeviceType plus an ordinal index.

View source ↗Full module reference on javadoc.io ↗

Record components

type
DeviceType
index
int

Methods

static Device cpu()
static Device cuda()
static Device cuda(int index)
static Device rocm()
static Device rocm(int index)
static Device vulkan()
static Device vulkan(int index)
String toString()