tvm
Public Member Functions | Static Public Member Functions | Friends | List of all members
tvm::TargetTagRegEntry Class Reference

#include <tag.h>

Collaboration diagram for tvm::TargetTagRegEntry:

Public Member Functions

TargetTagRegEntryset_config (Map< String, ObjectRef > config)
 Set the config dict corresponding to the target tag. More...
 
TargetTagRegEntrywith_config (String key, ObjectRef value)
 Add a key-value pair to the config dict. More...
 
TargetTagRegEntryset_name ()
 Set name of the TargetTag to be the same as registry if it is empty. More...
 

Static Public Member Functions

static TargetTagRegEntryRegisterOrGet (const String &target_tag_name)
 Register or get a new entry. More...
 

Friends

template<typename , typename >
class AttrRegistry
 
class TargetTag
 

Member Function Documentation

◆ RegisterOrGet()

static TargetTagRegEntry& tvm::TargetTagRegEntry::RegisterOrGet ( const String target_tag_name)
static

Register or get a new entry.

Parameters
target_tag_nameThe name of the TargetTag.
Returns
the corresponding entry.

◆ set_config()

TargetTagRegEntry & tvm::TargetTagRegEntry::set_config ( Map< String, ObjectRef config)
inline

Set the config dict corresponding to the target tag.

Parameters
configThe config dict for target creation

◆ set_name()

TargetTagRegEntry & tvm::TargetTagRegEntry::set_name ( )
inline

Set name of the TargetTag to be the same as registry if it is empty.

◆ with_config()

TargetTagRegEntry & tvm::TargetTagRegEntry::with_config ( String  key,
ObjectRef  value 
)
inline

Add a key-value pair to the config dict.

Parameters
keyThe attribute name
valueThe attribute value

Friends And Related Function Documentation

◆ AttrRegistry

template<typename , typename >
friend class AttrRegistry
friend

◆ TargetTag

friend class TargetTag
friend

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