Layout is to describe how data is organized within an N-dimention tensor. It is composed of upper cases, lower cases and numbers, where upper case indicates a primal axis and the corresponding lower case with factor size indicates the subordinate axis. For example, NCHW16c can describe a 5-D tensor of [batch_size, channel, height, width, channel_block]. Here subordinate axis channel_block=16 is the factor size of the primal axis C (channel). Layout for scalar is defined, while both its name and axes have size 0.
More...
#include <data_layout.h>
|
String | name |
| string representation of layout, "" for scalar. More...
|
|
Array< tir::IterVar > | axes |
| specify each axis of the layout, in which the variable name is the name of the axis. The IterVar's extent indicates the size of the axis, it is a variable for a primal axis, but a constant for a subordinate axis. Empty for scalar's layout. More...
|
|
|
static constexpr const char * | _type_key = "tir.Layout" |
|
Layout is to describe how data is organized within an N-dimention tensor. It is composed of upper cases, lower cases and numbers, where upper case indicates a primal axis and the corresponding lower case with factor size indicates the subordinate axis. For example, NCHW16c can describe a 5-D tensor of [batch_size, channel, height, width, channel_block]. Here subordinate axis channel_block=16 is the factor size of the primal axis C (channel). Layout for scalar is defined, while both its name and axes have size 0.
◆ RegisterReflection()
static void tvm::tir::LayoutNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
tvm::tir::LayoutNode::TVM_DECLARE_FINAL_OBJECT_INFO |
( |
LayoutNode |
, |
|
|
Object |
|
|
) |
| |
◆ _type_key
constexpr const char* tvm::tir::LayoutNode::_type_key = "tir.Layout" |
|
staticconstexpr |
◆ axes
specify each axis of the layout, in which the variable name is the name of the axis. The IterVar's extent indicates the size of the axis, it is a variable for a primal axis, but a constant for a subordinate axis. Empty for scalar's layout.
◆ name
String tvm::tir::LayoutNode::name |
string representation of layout, "" for scalar.
The documentation for this class was generated from the following file: