|
tvm
|
Fuse multiple iterators by summing them with scaling. More...
#include <iter_affine_map.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("arith.IterSumExpr", IterSumExprNode, IterMapExprNode) | |
Public Member Functions inherited from tvm::arith::IterMapExprNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("arith.IterMapExpr", IterMapExprNode, PrimExprNode) | |
Public Member Functions inherited from tvm::PrimExprNode | |
| TVM_OBJECT_ENABLE_SCRIPT_PRINTER () | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExpr", PrimExprNode, BaseExprNode) | |
Public Member Functions inherited from tvm::BaseExprNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", BaseExprNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::PrimExprNode | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::BaseExprNode | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::Array< IterSplitExpr > | args |
| The args to the sum. More... | |
| PrimExpr | base |
| The base offset. More... | |
Public Attributes inherited from tvm::PrimExprNode | |
| DataType | dtype |
| The runtime data type of the primitive expression. More... | |
Public Attributes inherited from tvm::BaseExprNode | |
| 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 Public Attributes inherited from tvm::arith::IterMapExprNode | |
| static constexpr const uint32_t | _type_child_slots = 2 |
Static Public Attributes inherited from tvm::PrimExprNode | |
| static constexpr const uint32_t | _type_child_slots = 40 |
Static Public Attributes inherited from tvm::BaseExprNode | |
| 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_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "arith.IterSumExpr" | , |
| IterSumExprNode | , | ||
| IterMapExprNode | |||
| ) |
|
staticconstexpr |
| ffi::Array<IterSplitExpr> tvm::arith::IterSumExprNode::args |
The args to the sum.
| PrimExpr tvm::arith::IterSumExprNode::base |
The base offset.