mxnet
|
Base node of all primitive expressions. More...
#include <expr.h>
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 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" |
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.
mxnet::PrimExprNode::MXNET_DECLARE_BASE_OBJECT_INFO | ( | PrimExprNode | , |
BaseExprNode | |||
) |
|
static |
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.