tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::tir::BijectiveLayoutNode Class Reference

#include <data_layout.h>

Inheritance diagram for tvm::tir::BijectiveLayoutNode:
Collaboration diagram for tvm::tir::BijectiveLayoutNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (BijectiveLayoutNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

Array< PrimExprindex_forward_rule
 Describes how source axes can be mapped to the destination axes, e.g., [i0 / 16, i1, i0 % 16] can describe NC -> NC16n. More...
 
Array< PrimExprindex_backward_rule
 Describes how destination axes can be mapped to the source axes. More...
 
Array< PrimExprshape_forward_rule
 Describes how source shapes can be mapped to the destination shapes. More...
 
Array< PrimExprshape_backward_rule
 Describes how destination shapes can be mapped to the source shapes. More...
 
Layout src_layout
 The source layout. More...
 
Layout dst_layout
 The destination layout. More...
 

Static Public Attributes

static constexpr const char * _type_key = "tir.BijectiveLayout"
 

Member Function Documentation

◆ RegisterReflection()

static void tvm::tir::BijectiveLayoutNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::tir::BijectiveLayoutNode::TVM_DECLARE_FINAL_OBJECT_INFO ( BijectiveLayoutNode  ,
Object   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::tir::BijectiveLayoutNode::_type_key = "tir.BijectiveLayout"
staticconstexpr

◆ dst_layout

Layout tvm::tir::BijectiveLayoutNode::dst_layout

The destination layout.

◆ index_backward_rule

Array<PrimExpr> tvm::tir::BijectiveLayoutNode::index_backward_rule

Describes how destination axes can be mapped to the source axes.

◆ index_forward_rule

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

Array<PrimExpr> tvm::tir::BijectiveLayoutNode::shape_backward_rule

Describes how destination shapes can be mapped to the source shapes.

◆ shape_forward_rule

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 source layout.


The documentation for this class was generated from the following file: