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

Packages that use Tensor

Package

Description

Uses of Tensor in org.apache.tvm

Methods in org.apache.tvm that return Tensor

Modifier and Type

Method

Description

static Tensor

Tensor.empty(long[] shape)

Create an empty float32 array on cpu given shape.

static Tensor

Tensor.empty(long[] shape, Device dev)

Create an empty float32 array given shape and device.

static Tensor

Tensor.empty(long[] shape, TVMType dtype)

Create an empty array on cpu given shape and type.

static Tensor

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

Create an empty array given shape, type and device.