tvm
Classes | Public Member Functions | Static Public Member Functions | Friends | List of all members
tvm::relay::ExecutorRegEntry Class Reference

Helper structure to register Executors. More...

#include <executor.h>

Collaboration diagram for tvm::relay::ExecutorRegEntry:

Public Member Functions

template<typename ValueType >
ExecutorRegEntryadd_attr_option (const String &key)
 Register a valid configuration option and its ValueType for validation. More...
 
template<typename ValueType >
ExecutorRegEntryadd_attr_option (const String &key, ObjectRef default_value)
 Register a valid configuration option and its ValueType for validation. More...
 

Static Public Member Functions

static ExecutorRegEntryRegisterOrGet (const String &name)
 Register or get a new entry. More...
 

Friends

template<typename >
class AttrRegistryMapContainerMap
 
template<typename , typename >
class tvm::AttrRegistry
 
class Executor
 

Detailed Description

Helper structure to register Executors.

See also
TVM_REGISTER_EXECUTOR

Member Function Documentation

◆ add_attr_option() [1/2]

template<typename ValueType >
ExecutorRegEntry & tvm::relay::ExecutorRegEntry::add_attr_option ( const String key)
inline

Register a valid configuration option and its ValueType for validation.

Parameters
keyThe configuration key
Template Parameters
ValueTypeThe value type to be registered

◆ add_attr_option() [2/2]

template<typename ValueType >
ExecutorRegEntry & tvm::relay::ExecutorRegEntry::add_attr_option ( const String key,
ObjectRef  default_value 
)
inline

Register a valid configuration option and its ValueType for validation.

Parameters
keyThe configuration key
default_valueThe default value of the key
Template Parameters
ValueTypeThe value type to be registered

◆ RegisterOrGet()

static ExecutorRegEntry& tvm::relay::ExecutorRegEntry::RegisterOrGet ( const String name)
static

Register or get a new entry.

Parameters
nameThe name of the operator.
Returns
the corresponding entry.

Friends And Related Function Documentation

◆ AttrRegistryMapContainerMap

template<typename >
friend class AttrRegistryMapContainerMap
friend

◆ Executor

friend class Executor
friend

◆ tvm::AttrRegistry

template<typename , typename >
friend class tvm::AttrRegistry
friend

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