Uses of Class org.apache.tvm.NDArray (TVM4J Package - Core 0.0.1-SNAPSHOT API)

Packages that use NDArray

Package

Description

Uses of NDArray in org.apache.tvm

Methods in org.apache.tvm that return NDArray

Modifier and Type

Method

Description

static NDArray

NDArray.empty(long[] shape)

Create an empty float32 array on cpu given shape.

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)

Create an empty array on cpu given shape and type.

static NDArray

NDArray.empty(long[] shape, TVMType dtype, Device dev)

Create an empty array given shape, type and device.