tvm
|
Describes one parameter that should be linked into the generated module. More...
#include <function.h>
Public Member Functions | |
void | VisitAttrs (tvm::AttrVisitor *v) |
TVM_DECLARE_FINAL_OBJECT_INFO (LinkedParamNode, 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 | |
int64_t | id |
Unique numeric identifier used by runtimes to lookup this parameter. More... | |
::tvm::runtime::NDArray | param |
Parameter data which should get linked into the final module. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "tir.LinkedParam" |
![]() | |
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 |
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... | |
Describes one parameter that should be linked into the generated module.
When parameters are to be linked in with generated code (i.e. on target_host-compatible backends), Relay attaches instances of this object to a global TIR function. Code-generators use the information contained in this node to include the parameter data in the generated module.
tvm::tir::LinkedParamNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | LinkedParamNode | , |
Object | |||
) |
|
inline |
|
static |
int64_t tvm::tir::LinkedParamNode::id |
Unique numeric identifier used by runtimes to lookup this parameter.
::tvm::runtime::NDArray tvm::tir::LinkedParamNode::param |
Parameter data which should get linked into the final module.