Uses of Class
org.apache.tvm.TVMContext

Packages that use TVMContext

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

Uses of TVMContext in org.apache.tvm

Methods in org.apache.tvm that return TVMContext

Modifier and Type	Method and Description
static TVMContext	TVMContext.cpu()
static TVMContext	TVMContext.cpu(int devId) Construct a CPU device.
TVMContext	NDArray.ctx() Get the context of current array.
static TVMContext	TVMContext.gpu()
static TVMContext	TVMContext.gpu(int devId) Construct a GPU device.
static TVMContext	TVMContext.hexagon()
static TVMContext	TVMContext.hexagon(int devId) Construct a Hexagon device.
static TVMContext	TVMContext.metal()
static TVMContext	TVMContext.metal(int devId) Construct a metal device.
static TVMContext	TVMContext.opencl()
static TVMContext	TVMContext.opencl(int devId) Construct a OpenCL device.
static TVMContext	TVMContext.vpi()
static TVMContext	TVMContext.vpi(int devId) Construct a VPI simulated device.
static TVMContext	TVMContext.vulkan()
static TVMContext	TVMContext.vulkan(int devId) Construct a Vulkan device.

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

Modifier and Type	Method and Description
static NDArray	NDArray.empty(long[] shape, TVMContext ctx) Create an empty float32 array given shape and device.
static NDArray	NDArray.empty(long[] shape, TVMType dtype, TVMContext ctx) Create an empty array given shape, type and device.

Uses of TVMContext in org.apache.tvm.contrib

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

Modifier and Type	Method and Description
static GraphModule	GraphRuntime.create(String graphJson, Module libmod, TVMContext ctx) Create a runtime executor module given a graph and module.

Uses of TVMContext in org.apache.tvm.rpc

Subclasses of TVMContext in org.apache.tvm.rpc

Modifier and Type	Class and Description
class	TVMRemoteContext

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

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