tvm::s_tir::meta_schedule::TensorInfo Class Reference

Managed reference to TensorInfoObj. More...

#include <arg_info.h>

Inheritance diagram for tvm::s_tir::meta_schedule::TensorInfo:

Collaboration diagram for tvm::s_tir::meta_schedule::TensorInfo:

Public Member Functions
	TensorInfo (DLDataType dtype, ffi::Shape shape)
	Constructor of TensorInfo.
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (TensorInfo, ArgInfo, TensorInfoObj)
Public Member Functions inherited from tvm::s_tir::meta_schedule::ArgInfo
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (ArgInfo, ffi::ObjectRef, ArgInfoObj)

Static Public Member Functions
static TensorInfo	FromJSON (const ffi::ObjectRef &json_obj)
	Parse the argument information from a JSON object.
Static Public Member Functions inherited from tvm::s_tir::meta_schedule::ArgInfo
static ArgInfo	FromJSON (const ffi::ObjectRef &json_obj)
	Parse the argument information from a JSON object.
static ffi::Array< ArgInfo, void >	FromFunction (const tirx::Function &func)
	Extract a list of the argument information from Function.
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.

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

Detailed Description

Managed reference to TensorInfoObj.

See also

TensorInfoObj

Constructor & Destructor Documentation

TensorInfo()

tvm::s_tir::meta_schedule::TensorInfo::TensorInfo ( DLDataType  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::s_tir::meta_schedule::TensorInfo::FromJSON ( const ffi::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::s_tir::meta_schedule::TensorInfo::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE	(	TensorInfo	,
		ArgInfo	,
		TensorInfoObj
	)

---

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

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