Uses of Class
org.apache.tvm.Device

Packages that use Device

Package	Description
org.apache.tvm
org.apache.tvm.contrib
org.apache.tvm.rpc

Uses of Device in org.apache.tvm

Methods in org.apache.tvm that return Device

Modifier and Type	Method and 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	NDArray.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 and Description
static NDArray	NDArray.empty(long[] shape, Device dev) Create an empty float32 array given shape and device.
static NDArray	NDArray.empty(long[] shape, TVMType dtype, Device dev) Create an empty array given shape, type and device.

Uses of Device in org.apache.tvm.contrib

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

Modifier and Type	Method and Description
static GraphModule	GraphExecutor.create(String graphJson, Module libmod, Device dev) Create a runtime executor module given a graph and module.

Uses of Device in org.apache.tvm.rpc

Subclasses of Device in org.apache.tvm.rpc

Modifier and Type	Class and Description
class	TVMRemoteDevice

Methods in org.apache.tvm.rpc that return Device

Modifier and Type	Method and Description
Device	RPCSession.cl() Construct remote OpenCL device.
Device	RPCSession.cl(int devId) Construct remote OpenCL device.
Device	RPCSession.cpu() Construct remote CPU device.
Device	RPCSession.cpu(int devId) Construct remote CPU device.
Device	RPCSession.cuda() Construct remote CUDA GPU device.
Device	RPCSession.cuda(int devId) Construct remote CUDA GPU device.
Device	RPCSession.device(int devType) Construct a remote device.
Device	RPCSession.device(int devType, int devId) Construct a remote device.
Device	RPCSession.device(String devType) Construct a remote device.
Device	RPCSession.device(String devType, int devId) Construct a remote device.
Device	RPCSession.metal() Construct remote Metal device.
Device	RPCSession.metal(int devId) Construct remote Metal device.
Device	RPCSession.vulkan() Construct remote OpenCL device.
Device	RPCSession.vulkan(int devId) Construct remote OpenCL device.