类的使用
org.apache.tvm.Device

使用 Device 的包

包	描述
org.apache.tvm
org.apache.tvm.rpc

org.apache.tvm 中 Device 的使用

org.apache.tvm 中返回 Device 的方法

修饰符和类型	方法	描述
static Device	Device.cpu()
static Device	Device.cpu(int devId)	构造一个 CPU 设备。
static Device	Device.cuda()
static Device	Device.cuda(int devId)	构造一个 CUDA GPU 设备。
Device	NDArray.device()	获取当前数组的设备。
static Device	Device.hexagon()
static Device	Device.hexagon(int devId)	构造一个 Hexagon 设备。
static Device	Device.metal()
static Device	Device.metal(int devId)	构造一个 metal 设备。
static Device	Device.opencl()
static Device	Device.opencl(int devId)	构造一个 OpenCL 设备。
static Device	Device.vpi()
static Device	Device.vpi(int devId)	构造一个 VPI 模拟设备。
static Device	Device.vulkan()
static Device	Device.vulkan(int devId)	构造一个 Vulkan 设备。

org.apache.tvm 中参数类型为 Device 的方法

修饰符和类型	方法	描述
static NDArray	NDArray.empty(long[] shape, Device dev)	根据形状和设备创建一个空的 float32 数组。
static NDArray	NDArray.empty(long[] shape, TVMType dtype, Device dev)	根据形状、类型和设备创建一个空数组。
Function	Function.pushArg(Device arg)	将参数推送到函数。

org.apache.tvm.rpc 中 Device 的使用

org.apache.tvm.rpc 中 Device 的子类

修饰符和类型	类	描述
类	TVMRemoteDevice

org.apache.tvm.rpc 中返回 Device 的方法

修饰符和类型	方法	描述
Device	RPCSession.cl()	构造远程 OpenCL 设备。
Device	RPCSession.cl(int devId)	构造远程 OpenCL 设备。
Device	RPCSession.cpu()	构造远程 CPU 设备。
Device	RPCSession.cpu(int devId)	构造远程 CPU 设备。
Device	RPCSession.cuda()	构造远程 CUDA GPU 设备。
Device	RPCSession.cuda(int devId)	构造远程 CUDA GPU 设备。
Device	RPCSession.device(int devType)	构造一个远程设备。
Device	RPCSession.device(int devType, int devId)	构造一个远程设备。
Device	RPCSession.device(String devType)	构造一个远程设备。
Device	RPCSession.device(String devType, int devId)	构造一个远程设备。
Device	RPCSession.metal()	构造远程 Metal 设备。
Device	RPCSession.metal(int devId)	构造远程 Metal 设备。
Device	RPCSession.vulkan()	构造远程 OpenCL 设备。
Device	RPCSession.vulkan(int devId)	构造远程 OpenCL 设备。