Uses of Class
org.apache.tvm.NDArray

Packages that use NDArray

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

Uses of NDArray in org.apache.tvm

Methods in org.apache.tvm that return NDArray

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

Uses of NDArray in org.apache.tvm.contrib

Methods in org.apache.tvm.contrib that return NDArray

Modifier and Type	Method and Description
NDArray	GraphModule.debugGetOutput(int node, NDArray out) Run graph up to node and get the output to out.
NDArray	GraphModule.debugGetOutput(String node, NDArray out) Run graph up to node and get the output to out.
NDArray	GraphModule.getInput(int index, NDArray out) Get index-th input to out.
NDArray	GraphModule.getOutput(int index, NDArray out) Get index-th output to out.

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

Modifier and Type	Method and Description
NDArray	GraphModule.debugGetOutput(int node, NDArray out) Run graph up to node and get the output to out.
NDArray	GraphModule.debugGetOutput(String node, NDArray out) Run graph up to node and get the output to out.
NDArray	GraphModule.getInput(int index, NDArray out) Get index-th input to out.
NDArray	GraphModule.getOutput(int index, NDArray out) Get index-th output to out.
GraphModule	GraphModule.setInput(int key, NDArray value) Set inputs to the module.
GraphModule	GraphModule.setInput(String key, NDArray value) Set inputs to the module.