tvm
Loading...
Searching...
No Matches
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 (ffi::Map< ffi::String, Any > config)
 Set the config dict corresponding to the target tag.
 
TargetTagRegEntrywith_config (ffi::String key, Any value)
 Add a key-value pair to the config dict.
 
TargetTagRegEntryset_name ()
 Set name of the TargetTag to be the same as registry if it is empty.
 

Static Public Member Functions

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

Friends

template<typename , typename >
class AttrRegistry
 
class TargetTag
 

Member Function Documentation

◆ RegisterOrGet()

static TargetTagRegEntry & tvm::TargetTagRegEntry::RegisterOrGet ( const ffi::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 ( ffi::Map< ffi::String, Any 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 ( ffi::String  key,
Any  value 
)
inline

Add a key-value pair to the config dict.

Parameters
keyThe attribute name
valueThe attribute value

Friends And Related Symbol Documentation

◆ AttrRegistry

template<typename , typename >
friend class AttrRegistry
friend

◆ TargetTag


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