tvm::s_tir::meta_schedule::TensorInfoObj Class Reference

The tensor argument information. More...

#include <arg_info.h>

Inheritance diagram for tvm::s_tir::meta_schedule::TensorInfoObj:

Collaboration diagram for tvm::s_tir::meta_schedule::TensorInfoObj:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.meta_schedule.TensorInfo", TensorInfoObj, ArgInfoObj)
ffi::ObjectRef	AsJSON () const
	Converts the ArgInfo to its corresponding JSON representation.
Public Member Functions inherited from tvm::s_tir::meta_schedule::ArgInfoObj
	TVM_FFI_DECLARE_OBJECT_INFO ("s_tir.meta_schedule.ArgInfo", ArgInfoObj, ffi::Object)
virtual	~ArgInfoObj ()=default
	Default destructor.

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
DLDataType	dtype
	The data type of the tensor.
ffi::Shape	shape
	The shape of the tensor.

Detailed Description

The tensor argument information.

Member Function Documentation

AsJSON()

ffi::ObjectRef tvm::s_tir::meta_schedule::TensorInfoObj::AsJSON ( ) const

virtual

Converts the ArgInfo to its corresponding JSON representation.

Implements tvm::s_tir::meta_schedule::ArgInfoObj.

RegisterReflection()

static void tvm::s_tir::meta_schedule::TensorInfoObj::RegisterReflection ( )

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::meta_schedule::TensorInfoObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.meta_schedule.TensorInfo"	,
		TensorInfoObj	,
		ArgInfoObj
	)

Member Data Documentation

dtype

DLDataType tvm::s_tir::meta_schedule::TensorInfoObj::dtype

The data type of the tensor.

shape

ffi::Shape tvm::s_tir::meta_schedule::TensorInfoObj::shape

The shape of the tensor.

---

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

• include/tvm/s_tir/meta_schedule/arg_info.h