tvm::meta_schedule::ArgInfo Class Reference

Managed reference to ArgInfoNode. More...

#include <arg_info.h>

Inheritance diagram for tvm::meta_schedule::ArgInfo:

Collaboration diagram for tvm::meta_schedule::ArgInfo:

Public Member Functions
	TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS (ArgInfo, runtime::ObjectRef, ArgInfoNode)

Static Public Member Functions
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...

Protected Member Functions
	ArgInfo ()=default

Detailed Description

Managed reference to ArgInfoNode.

See also

ArgInfoNode

Constructor & Destructor Documentation

ArgInfo()

tvm::meta_schedule::ArgInfo::ArgInfo ( )

protecteddefault

Member Function Documentation

FromEntryFunc()

static Array<ArgInfo, void> tvm::meta_schedule::ArgInfo::FromEntryFunc ( const IRModule &  mod, bool  remove_preproc  )

static

Extract a list of the argument information from the entry func of an IRModule.

Parameters

mod	The IRModule to extract argument information from.
remove_preproc	Whether to remove the preprocessing blocks.

Returns

An array of the argument information derived.

FromJSON()

static ArgInfo tvm::meta_schedule::ArgInfo::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.

FromPrimFunc()

static Array<ArgInfo, void> tvm::meta_schedule::ArgInfo::FromPrimFunc ( const tir::PrimFunc &  func )

static

Extract a list of the argument information from PrimFunc.

Parameters

func	The PrimFunc to get argument information from.

Returns

An array of the argument information derived.

TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS()

tvm::meta_schedule::ArgInfo::TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS	(	ArgInfo	,
		runtime::ObjectRef	,
		ArgInfoNode
	)

---

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

• include/tvm/meta_schedule/arg_info.h