Managed reference to TensorInfoNode. More...

#include <arg_info.h>

Inheritance diagram for tvm::meta_schedule::TensorInfo:

Collaboration diagram for tvm::meta_schedule::TensorInfo:

Public Member Functions
	TensorInfo (runtime::DataType dtype, ffi::Shape shape)
	Constructor of TensorInfo. More...

	TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS (TensorInfo, ArgInfo, TensorInfoNode)

Public Member Functions inherited from tvm::meta_schedule::ArgInfo
	TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS (ArgInfo, runtime::ObjectRef, ArgInfoNode)

Static Public Member Functions
static TensorInfo	FromJSON (const ObjectRef &json_obj)
	Parse the argument information from a JSON object. More...

Static Public Member Functions inherited from tvm::meta_schedule::ArgInfo
static ArgInfo	FromJSON (const ObjectRef &json_obj)
	Parse the argument information from a JSON object. More...

static Array< ArgInfo, void >	FromPrimFunc (const tir::PrimFunc &func)
	Extract a list of the argument information from PrimFunc. More...

static Array< ArgInfo, void >	FromEntryFunc (const IRModule &mod, bool remove_preproc)
	Extract a list of the argument information from the entry func of an IRModule. More...

Additional Inherited Members
Protected Member Functions inherited from tvm::meta_schedule::ArgInfo
	ArgInfo ()=default

Detailed Description

Managed reference to TensorInfoNode.

Constructor & Destructor Documentation

tvm::meta_schedule::TensorInfo::TensorInfo	(	runtime::DataType	dtype,
		ffi::Shape	shape
	)

explicit

Constructor of TensorInfo.

Parameters

dtype	The data type of the tensor argument.
shape	The shape tuple of the tensor argument.

static TensorInfo tvm::meta_schedule::TensorInfo::FromJSON ( const ObjectRef & json_obj )

static

Parse the argument information from a JSON object.

Parameters

json_obj The json object to parse.

tvm::meta_schedule::TensorInfo::TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS	(	TensorInfo	,
		ArgInfo	,
		TensorInfoNode
	)

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