Uses of Class
org.apache.tvm.Device

Packages that use Device

Package	Description
org.apache.tvm

Uses of Device in org.apache.tvm

Methods in org.apache.tvm that return Device

Modifier and Type	Method	Description
static Device	Device.cpu()
static Device	Device.cpu(int devId)	Construct a CPU device.
static Device	Device.cuda()
static Device	Device.cuda(int devId)	Construct a CUDA GPU device.
Device	Tensor.device()	Get the device of current array.
static Device	Device.hexagon()
static Device	Device.hexagon(int devId)	Construct a Hexagon device.
static Device	Device.metal()
static Device	Device.metal(int devId)	Construct a metal device.
static Device	Device.opencl()
static Device	Device.opencl(int devId)	Construct a OpenCL device.
static Device	Device.vpi()
static Device	Device.vpi(int devId)	Construct a VPI simulated device.
static Device	Device.vulkan()
static Device	Device.vulkan(int devId)	Construct a Vulkan device.

Methods in org.apache.tvm with parameters of type Device

Modifier and Type	Method	Description
static Tensor	Tensor.empty(long[] shape, Device dev)	Create an empty float32 array given shape and device.
static Tensor	Tensor.empty(long[] shape, TVMType dtype, Device dev)	Create an empty array given shape, type and device.
Function	Function.pushArg(Device arg)	Push argument to the function.