tag.h File Reference

Target tag registry. More...

#include <tvm/node/attr_registry_map.h>
#include <tvm/node/node.h>
#include <tvm/target/target.h>
#include <utility>

Include dependency graph for tag.h:

Go to the source code of this file.

Classes
class	tvm::TargetTagNode
	A target tag. More...
class	tvm::TargetTag
	Managed reference class to TargetTagNode. More...
class	tvm::TargetTagRegEntry

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.

Macros
#define	TVM_TARGET_TAG_REGISTER_VAR_DEF    static DMLC_ATTRIBUTE_UNUSED ::tvm::TargetTagRegEntry& __make_##TargetTag
#define	TVM_REGISTER_TARGET_TAG(TargetTagName)
	Register a new target tag, or set attribute of the corresponding target tag. More...

Detailed Description

Target tag registry.

Macro Definition Documentation

TVM_REGISTER_TARGET_TAG

#define TVM_REGISTER_TARGET_TAG

(

TargetTagName

)

Value:

  TVM_STR_CONCAT(TVM_TARGET_TAG_REGISTER_VAR_DEF, __COUNTER__) = \
      ::tvm::TargetTagRegEntry::RegisterOrGet(TargetTagName).set_name()

Register a new target tag, or set attribute of the corresponding target tag.

Parameters

TargetTagName

The name of target tag

TVM_TARGET_TAG_REGISTER_VAR_DEF

#define TVM_TARGET_TAG_REGISTER_VAR_DEF    static DMLC_ATTRIBUTE_UNUSED ::tvm::TargetTagRegEntry& __make_##TargetTag