tvm
|
Fuse multiple iterators by summing them with scaling. More...
#include <iter_affine_map.h>
Public Member Functions | |
TVM_DECLARE_FINAL_OBJECT_INFO (IterSumExprNode, IterMapExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (IterMapExprNode, PrimExprNode) | |
![]() | |
TVM_OBJECT_ENABLE_SCRIPT_PRINTER () | |
TVM_DECLARE_BASE_OBJECT_INFO (PrimExprNode, BaseExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseExprNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
Array< IterSplitExpr > | args |
The args to the sum. More... | |
PrimExpr | base |
The base offset. More... | |
![]() | |
DataType | dtype |
The runtime data type of the primitive expression. 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 = kTVMFFISEqHashKindTreeNode |
static constexpr const char * | _type_key = "arith.IterSumExpr" |
![]() | |
static constexpr const char * | _type_key = "arith.IterMapExpr" |
static constexpr const uint32_t | _type_child_slots = 2 |
![]() | |
static constexpr const char * | _type_key = "ir.PrimExpr" |
static constexpr const uint32_t | _type_child_slots = 40 |
![]() | |
static constexpr const char * | _type_key = "ir.BaseExpr" |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
static constexpr const uint32_t | _type_child_slots = 64 |
Fuse multiple iterators by summing them with scaling.
result = sum(args) + base
|
inlinestatic |
tvm::arith::IterSumExprNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | IterSumExprNode | , |
IterMapExprNode | |||
) |
|
staticconstexpr |
|
staticconstexpr |
Array<IterSplitExpr> tvm::arith::IterSumExprNode::args |
The args to the sum.
PrimExpr tvm::arith::IterSumExprNode::base |
The base offset.