tvm
Public Member Functions | Static Public Member Functions | Protected Member Functions | List of all members
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)
 
- Public Member Functions inherited from tvm::runtime::ObjectRef
 ObjectRef ()=default
 default constructor More...
 
 ObjectRef (ObjectPtr< Object > data)
 Constructor from existing object ptr. More...
 
bool same_as (const ObjectRef &other) const
 Comparator. More...
 
bool operator== (const ObjectRef &other) const
 Comparator. More...
 
bool operator!= (const ObjectRef &other) const
 Comparator. More...
 
bool operator< (const ObjectRef &other) const
 Comparator. More...
 
bool defined () const
 
const Objectget () const
 
const Objectoperator-> () const
 
bool unique () const
 
int use_count () const
 
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType * as () const
 Try to downcast the internal Object to a raw pointer of a corresponding type. More...
 
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType > as () const
 Try to downcast the ObjectRef to a Optional<T> of the requested type. More...
 

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
 
- Protected Member Functions inherited from tvm::runtime::ObjectRef
Objectget_mutable () const
 

Additional Inherited Members

- Public Types inherited from tvm::runtime::ObjectRef
using ContainerType = Object
 type indicate the container type. More...
 
- Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool _type_is_nullable = true
 
- Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T DowncastNoCheck (ObjectRef ref)
 Internal helper function downcast a ref without check. More...
 
static void FFIClearAfterMove (ObjectRef *ref)
 Clear the object ref data field without DecRef after we successfully moved the field. More...
 
template<typename ObjectType >
static ObjectPtr< ObjectType > GetDataPtr (const ObjectRef &ref)
 Internal helper function get data_ as ObjectPtr of ObjectType. More...
 
- Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Objectdata_
 Internal pointer that backs the reference. More...
 

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
modThe IRModule to extract argument information from.
remove_preprocWhether 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_objThe 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
funcThe 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: