tvm
|
Compilation target. More...
#include <target.h>
Public Member Functions | |
const std::string & | str () const |
The raw string representation of the target. More... | |
Map< String, ObjectRef > | Export () const |
Optional< Target > | GetHost () const |
String | ToDebugString () const |
Returns a human readable representation of Target which includes all fields, especially the host. Useful for diagnostic messages and debugging. More... | |
void | VisitAttrs (AttrVisitor *v) |
template<typename TObjectRef > | |
Optional< TObjectRef > | GetAttr (const std::string &attr_key, Optional< TObjectRef > default_value=Optional< TObjectRef >(nullptr)) const |
Get an entry from attrs of the target. More... | |
template<typename TObjectRef > | |
Optional< TObjectRef > | GetAttr (const std::string &attr_key, TObjectRef default_value) const |
Get an entry from attrs of the target. More... | |
std::vector< std::string > | GetKeys () const |
Get the keys for this target as a vector of string. More... | |
std::unordered_set< std::string > | GetLibs () const |
Get the keys for this target as an unordered_set of string. More... | |
bool | SEqualReduce (const TargetNode *other, SEqualReducer equal) const |
void | SHashReduce (SHashReducer hash_reduce) const |
TVM_DECLARE_FINAL_OBJECT_INFO (TargetNode, Object) | |
![]() | |
uint32_t | type_index () const |
std::string | GetTypeKey () const |
size_t | GetTypeKeyHash () const |
template<typename TargetType > | |
bool | IsInstance () const |
bool | unique () const |
Object () | |
Object (const Object &other) | |
Object (Object &&other) | |
Object & | operator= (const Object &other) |
Object & | operator= (Object &&other) |
Public Attributes | |
TargetKind | kind |
The kind of the target device. More... | |
Optional< ObjectRef > | host |
Target host information, must be Target type. More... | |
String | tag |
Tag of the the target, can be empty. More... | |
Array< String > | keys |
Keys for this target. More... | |
Map< String, ObjectRef > | attrs |
Collection of attributes. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "Target" |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
![]() | |
static constexpr const char * | _type_key = "runtime.Object" |
static constexpr bool | _type_final = false |
static constexpr uint32_t | _type_child_slots = 0 |
static constexpr bool | _type_child_slots_can_overflow = true |
static constexpr bool | _type_has_method_visit_attrs = true |
static constexpr bool | _type_has_method_sequal_reduce = false |
static constexpr bool | _type_has_method_shash_reduce = false |
static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Friends | |
class | TargetInternal |
Additional Inherited Members | |
![]() | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
![]() | |
static std::string | TypeIndex2Key (uint32_t tindex) |
Get the type key of the corresponding index from runtime. More... | |
static size_t | TypeIndex2KeyHash (uint32_t tindex) |
Get the type key hash of the corresponding index from runtime. More... | |
static uint32_t | TypeKey2Index (const std::string &key) |
Get the type index of the corresponding key from runtime. More... | |
static uint32_t | _GetOrAllocRuntimeTypeIndex () |
static uint32_t | RuntimeTypeIndex () |
![]() | |
void | IncRef () |
developer function, increases reference counter. More... | |
void | DecRef () |
developer function, decrease reference counter. More... | |
![]() | |
static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
Get the type index using type key. More... | |
![]() | |
uint32_t | type_index_ {0} |
Type index(tag) that indicates the type of the object. More... | |
RefCounterType | ref_counter_ {0} |
The internal reference counter. More... | |
FDeleter | deleter_ = nullptr |
deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
Compilation target.
|
inline |
Get an entry from attrs of the target.
TObjectRef | Type of the attribute |
attr_key | The name of the attribute key |
default_value | The value returned if the key is not present |
|
inline |
Get an entry from attrs of the target.
TObjectRef | Type of the attribute |
attr_key | The name of the attribute key |
default_value | The value returned if the key is not present |
std::vector<std::string> tvm::TargetNode::GetKeys | ( | ) | const |
Get the keys for this target as a vector of string.
std::unordered_set<std::string> tvm::TargetNode::GetLibs | ( | ) | const |
Get the keys for this target as an unordered_set of string.
bool tvm::TargetNode::SEqualReduce | ( | const TargetNode * | other, |
SEqualReducer | equal | ||
) | const |
void tvm::TargetNode::SHashReduce | ( | SHashReducer | hash_reduce | ) | const |
const std::string& tvm::TargetNode::str | ( | ) | const |
The raw string representation of the target.
String tvm::TargetNode::ToDebugString | ( | ) | const |
Returns a human readable representation of Target
which includes all fields, especially the host. Useful for diagnostic messages and debugging.
TODO(mbs): The ReprPrinter version should perhaps switch to this form, however currently code depends on str() and << being the same.
tvm::TargetNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | TargetNode | , |
Object | |||
) |
|
inline |
|
friend |
|
static |
|
static |
|
static |
TargetKind tvm::TargetNode::kind |
The kind of the target device.
String tvm::TargetNode::tag |
Tag of the the target, can be empty.