tvm::meta_schedule::TensorInfo Class Reference

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_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (TensorInfo, ArgInfo, TensorInfoNode)
Public Member Functions inherited from tvm::meta_schedule::ArgInfo
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (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 ffi::Array< ArgInfo, void >	FromPrimFunc (const tir::PrimFunc &func)
	Extract a list of the argument information from PrimFunc. More...
static ffi::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.

See also

TensorInfoNode

Constructor & Destructor Documentation

TensorInfo()

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.

Member Function Documentation

FromJSON()

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.

Returns

The argument information parsed.

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

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

---

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

• include/tvm/meta_schedule/arg_info.h