tvm
|
Primitive Op(builtin intrinsics) More...
#include <op.h>
Public Member Functions | |
TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("ir.Op", OpNode, RelaxExprNode) | |
![]() | |
TVM_FFI_DECLARE_OBJECT_INFO ("ir.RelaxExpr", RelaxExprNode, BaseExprNode) | |
![]() | |
TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", BaseExprNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
ffi::String | name |
name of the operator More... | |
FuncType | op_type |
the type of the operator More... | |
ffi::String | description |
detailed description of the operator This can be used to generate docstring automatically for the operator. More... | |
ffi::Array< AttrFieldInfo > | arguments |
ffi::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... | |
![]() | |
ffi::Optional< ObjectRef > | struct_info_ = ffi::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 TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindUniqueInstance |
![]() | |
static constexpr const uint32_t | _type_child_slots = 22 |
![]() | |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
static constexpr const uint32_t | _type_child_slots = 64 |
Friends | |
template<typename > | |
class | AttrRegistryMapContainerMap |
template<typename , typename > | |
class | AttrRegistry |
class | OpRegEntry |
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.
|
inlinestatic |
tvm::OpNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "ir.Op" | , |
OpNode | , | ||
RelaxExprNode | |||
) |
|
friend |
|
friend |
|
friend |
|
staticconstexpr |
ffi::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.
ffi::String tvm::OpNode::attrs_type_key |
The type key of the attribute field This can be empty, in which case it defaults to anything.
ffi::String tvm::OpNode::description |
detailed description of the operator This can be used to generate docstring automatically for the operator.
ffi::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.