Describes one tensor argument to run_model. NOTE: while TIR allows for other types of arguments, such as scalars, the AOT run_model function does not currently accept these. Therefore it's not possible to express those in this metadata. A future patch may modify this. More...

#include <metadata_types.h>

Collaboration diagram for TVMTensorInfo:

Public Attributes
const char *	name
	Name of the tensor, as specified in the Relay program. More...

const int64_t *	shape
	Shape of the tensor. More...

int64_t	num_shape
	Rank of this tensor. More...

DLDataType	dtype
	Data type of one element of this tensor. More...

Detailed Description

Describes one tensor argument to run_model. NOTE: while TIR allows for other types of arguments, such as scalars, the AOT run_model function does not currently accept these. Therefore it's not possible to express those in this metadata. A future patch may modify this.

Member Data Documentation

◆ dtype

DLDataType TVMTensorInfo::dtype

Data type of one element of this tensor.

◆ name

const char* TVMTensorInfo::name

Name of the tensor, as specified in the Relay program.

◆ num_shape

int64_t TVMTensorInfo::num_shape

Rank of this tensor.

◆ shape

const int64_t* TVMTensorInfo::shape

Shape of the tensor.

The documentation for this struct was generated from the following file:

include/tvm/runtime/metadata_types.h