Base node of all primitive expressions. More...
#include <expr.h>
Inheritance diagram for mxnet::PrimExprNode:
Collaboration diagram for mxnet::PrimExprNode:
Public Member Functions | |
| MXNET_DECLARE_BASE_OBJECT_INFO (PrimExprNode, BaseExprNode) | |
 Public Member Functions inherited from mxnet::BaseExprNode | |
| MXNET_DECLARE_BASE_OBJECT_INFO (BaseExprNode, Object) | |
| Public Member Functions inherited from mxnet::runtime::Object | |
| uint32_t | type_index () const |
| std::string | GetTypeKey () const |
| size_t | GetTypeKeyHash () const |
| template<typename TargetType > | |
| bool | IsInstance () const |
| Object () | |
| Object (const Object &other) | |
| Object (Object &&other) | |
| Object & | operator= (const Object &other) |
| Object & | operator= (Object &&other) |
Public Attributes | |
| MXNetDataType | dtype |
| The runtime data type of the primitive expression. More... | |
Static Public Attributes | |
| static constexpr const char * | _type_key = "PrimExpr" |
| Static Public Attributes inherited from mxnet::BaseExprNode | |
| static constexpr const char * | _type_key = "Expr" |
| Static Public Attributes inherited from mxnet::runtime::Object | |
| static constexpr const char * | _type_key = "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 uint32_t | _type_index = TypeIndex::kDynamic |
Additional Inherited Members | |
| Public Types inherited from mxnet::runtime::Object | |
| typedef void(* | FDeleter) (Object *self) |
| Object deleter. More... | |
| using | RefCounterType = std::atomic< int32_t > |
| Static Public Member Functions inherited from mxnet::runtime::Object | |
| static MXNET_DLL std::string | TypeIndex2Key (uint32_t tindex) |
| Get the type key of the corresponding index from runtime. More... | |
| static MXNET_DLL size_t | TypeIndex2KeyHash (uint32_t tindex) |
| Get the type key hash of the corresponding index from runtime. More... | |
| static MXNET_DLL 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 () |
| Protected Member Functions inherited from mxnet::runtime::Object | |
| void | IncRef () |
| developer function, increases reference counter. More... | |
| void | DecRef () |
| developer function, decrease reference counter. More... | |
| Static Protected Member Functions inherited from mxnet::runtime::Object | |
| static MXNET_DLL 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... | |
| Protected Attributes inherited from mxnet::runtime::Object | |
| 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... | |
Detailed Description
Base node of all primitive expressions.
A primitive expression deals with low-level POD data types and handles without doing life-cycle management for objects.
PrimExpr is used in the low-level code optimizations and integer analysis.
- See also
- PrimExpr
Member Function Documentation
◆ MXNET_DECLARE_BASE_OBJECT_INFO()
| mxnet::PrimExprNode::MXNET_DECLARE_BASE_OBJECT_INFO | ( | PrimExprNode | , |
| BaseExprNode | |||
| ) |
Member Data Documentation
◆ _type_key
|
static |
◆ dtype
| MXNetDataType mxnet::PrimExprNode::dtype |
The runtime data type of the primitive expression.
MXNetDataType(dtype) provides coarse grained type information during compile time and runtime. It is eagerly built in PrimExpr expression construction and can be used for quick type checking.
dtype is sufficient to decide the Type of the PrimExpr when it corresponds to POD value types such as i32.
When dtype is MXNetDataType::Handle(), the expression could corresponds to a more fine-grained Type, and we can get the type by running lazy type inference.
The documentation for this class was generated from the following file:
- /work/mxnet/include/mxnet/ir/expr.h
