#include <data_layout.h>
◆ RegisterReflection()
static void tvm::tir::BijectiveLayoutNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
tvm::tir::BijectiveLayoutNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL |
( |
"tir.BijectiveLayout" |
, |
|
|
BijectiveLayoutNode |
, |
|
|
Object |
|
|
) |
| |
◆ dst_layout
Layout tvm::tir::BijectiveLayoutNode::dst_layout |
◆ index_backward_rule
ffi::Array<PrimExpr> tvm::tir::BijectiveLayoutNode::index_backward_rule |
Describes how destination axes can be mapped to the source axes.
◆ index_forward_rule
ffi::Array<PrimExpr> tvm::tir::BijectiveLayoutNode::index_forward_rule |
Describes how source axes can be mapped to the destination axes, e.g., [i0 / 16, i1, i0 % 16] can describe NC -> NC16n.
◆ shape_backward_rule
ffi::Array<PrimExpr> tvm::tir::BijectiveLayoutNode::shape_backward_rule |
Describes how destination shapes can be mapped to the source shapes.
◆ shape_forward_rule
ffi::Array<PrimExpr> tvm::tir::BijectiveLayoutNode::shape_forward_rule |
Describes how source shapes can be mapped to the destination shapes.
◆ src_layout
Layout tvm::tir::BijectiveLayoutNode::src_layout |
The documentation for this class was generated from the following file: