tvm
|
Primitive Op(builtin intrinsics) More...
#include <op.h>
Public Member Functions | |
void | VisitAttrs (AttrVisitor *v) |
bool | SEqualReduce (const OpNode *other, SEqualReducer equal) const |
void | SHashReduce (SHashReducer hash_reduce) const |
TVM_DECLARE_FINAL_OBJECT_INFO (OpNode, RelaxExprNode) | |
![]() | |
const Type & | checked_type () const |
template<typename TTypeNode > | |
const TTypeNode * | type_as () const |
Check if the inferred(checked) type of the Expr is backed by a TTypeNode and return it. More... | |
TVM_DECLARE_BASE_OBJECT_INFO (RelaxExprNode, BaseExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseExprNode, 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 | |
String | name |
name of the operator More... | |
FuncType | op_type |
the type of the operator More... | |
String | description |
detailed description of the operator This can be used to generate docstring automatically for the operator. More... | |
Array< AttrFieldInfo > | arguments |
String | attrs_type_key |
The type key of the attribute field This can be empty, in which case it defaults to anything. More... | |
uint32_t | attrs_type_index {0} |
attribute type index, this field varies in each run and is not exposed to frontend. More... | |
int32_t | num_inputs = -1 |
number of input arguments to the operator, -1 means it is variable length More... | |
int32_t | support_level = 10 |
support level of the operator, The lower the more priority it contains. This is in analogies to BLAS levels. More... | |
![]() | |
Type | checked_type_ = Type(nullptr) |
Stores the result of type inference(type checking). More... | |
Optional< ObjectRef > | struct_info_ = Optional<ObjectRef>() |
Stores the result of structure information of the expression that encapsulate both static shape and runtime information such as shape. More... | |
![]() | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "Op" |
![]() | |
static constexpr const char * | _type_key = "RelaxExpr" |
static constexpr const uint32_t | _type_child_slots = 22 |
![]() | |
static constexpr const char * | _type_key = "BaseExpr" |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
static constexpr const uint32_t | _type_child_slots = 64 |
![]() | |
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 | |
template<typename > | |
class | AttrRegistryMapContainerMap |
template<typename , typename > | |
class | AttrRegistry |
class | OpRegEntry |
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... | |
Primitive Op(builtin intrinsics)
This data structure stores the meta-data about primitive operators that can be invoked via Call.
Low-level IR intrinsics(such as libc.expf) are also implemented via Op.
|
inline |
|
inline |
tvm::OpNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | OpNode | , |
RelaxExprNode | |||
) |
|
inline |
|
friend |
|
friend |
|
friend |
|
staticconstexpr |
Array<AttrFieldInfo> tvm::OpNode::arguments |
uint32_t tvm::OpNode::attrs_type_index {0} |
attribute type index, this field varies in each run and is not exposed to frontend.
String tvm::OpNode::attrs_type_key |
The type key of the attribute field This can be empty, in which case it defaults to anything.
String tvm::OpNode::description |
detailed description of the operator This can be used to generate docstring automatically for the operator.
String tvm::OpNode::name |
name of the operator
int32_t tvm::OpNode::num_inputs = -1 |
number of input arguments to the operator, -1 means it is variable length
|
mutable |
the type of the operator
int32_t tvm::OpNode::support_level = 10 |
support level of the operator, The lower the more priority it contains. This is in analogies to BLAS levels.