tvm
|
Split of an iterator. More...
#include <iter_affine_map.h>
Public Member Functions | |
TVM_DECLARE_FINAL_OBJECT_INFO (IterSplitExprNode, 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 | |
IterMark | source |
The source marked iterator. More... | |
PrimExpr | lower_factor |
The lower factor to split the source. More... | |
PrimExpr | extent |
The extent of the split. More... | |
PrimExpr | scale |
Additional scale. 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.IterSplitExpr" |
![]() | |
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 |
Split of an iterator.
result = floormod(floordiv(source, lower_factor), extent) * scale
|
inlinestatic |
tvm::arith::IterSplitExprNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | IterSplitExprNode | , |
IterMapExprNode | |||
) |
|
staticconstexpr |
|
staticconstexpr |
PrimExpr tvm::arith::IterSplitExprNode::extent |
The extent of the split.
PrimExpr tvm::arith::IterSplitExprNode::lower_factor |
The lower factor to split the source.
PrimExpr tvm::arith::IterSplitExprNode::scale |
Additional scale.
IterMark tvm::arith::IterSplitExprNode::source |
The source marked iterator.