tvm::meta_schedule::TensorInfoNode Class Reference

The tensor argument information. More...

#include <arg_info.h>

Inheritance diagram for tvm::meta_schedule::TensorInfoNode:

Collaboration diagram for tvm::meta_schedule::TensorInfoNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (TensorInfoNode, ArgInfoNode)
ObjectRef	AsJSON () const
	Converts the ArgInfo to its corresponding JSON representation. More...
Public Member Functions inherited from tvm::meta_schedule::ArgInfoNode
	TVM_DECLARE_BASE_OBJECT_INFO (ArgInfoNode, runtime::Object)
virtual	~ArgInfoNode ()=default
	Default destructor. More...

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
runtime::DataType	dtype
	The data type of the tensor. More...
ffi::Shape	shape
	The shape of the tensor. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.TensorInfo"
Static Public Attributes inherited from tvm::meta_schedule::ArgInfoNode
static constexpr const char *	_type_key = "meta_schedule.ArgInfo"

Detailed Description

The tensor argument information.

Member Function Documentation

AsJSON()

ObjectRef tvm::meta_schedule::TensorInfoNode::AsJSON ( ) const

virtual

Converts the ArgInfo to its corresponding JSON representation.

Implements tvm::meta_schedule::ArgInfoNode.

RegisterReflection()

static void tvm::meta_schedule::TensorInfoNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::TensorInfoNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	TensorInfoNode	,
		ArgInfoNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::meta_schedule::TensorInfoNode::_type_key = "meta_schedule.TensorInfo"

staticconstexpr

dtype

runtime::DataType tvm::meta_schedule::TensorInfoNode::dtype

The data type of the tensor.

shape

ffi::Shape tvm::meta_schedule::TensorInfoNode::shape

The shape of the tensor.

---

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

• include/tvm/meta_schedule/arg_info.h