tvm
Public Member Functions | List of all members
tvm::tir::BijectiveLayout Class Reference

Bijective function mapping for data layout transformation. Given two Layout, BijectiveLayout build and store the mapping rules, provides API to transform N-dimention tensor from the source indices (i0, i1, .., im) to the destination indices (j0, j1, .., jm). More...

#include <data_layout.h>

Inheritance diagram for tvm::tir::BijectiveLayout:
Collaboration diagram for tvm::tir::BijectiveLayout:

Public Member Functions

 BijectiveLayout (Layout src_layout, Layout dst_layout)
 The constructor. More...
 
Array< PrimExprForwardShape (const Array< PrimExpr > &shape) const
 
Array< PrimExprBackwardShape (const Array< PrimExpr > &dst_shape) const
 
Array< PrimExprForwardIndex (const Array< PrimExpr > &index) const
 
Array< PrimExprBackwardIndex (const Array< PrimExpr > &dst_index) const
 
 TVM_DEFINE_OBJECT_REF_METHODS (BijectiveLayout, ObjectRef, BijectiveLayoutNode)
 

Detailed Description

Bijective function mapping for data layout transformation. Given two Layout, BijectiveLayout build and store the mapping rules, provides API to transform N-dimention tensor from the source indices (i0, i1, .., im) to the destination indices (j0, j1, .., jm).

Constructor & Destructor Documentation

◆ BijectiveLayout()

tvm::tir::BijectiveLayout::BijectiveLayout ( Layout  src_layout,
Layout  dst_layout 
)

The constructor.

Parameters
src_layoutThe source layout
dst_layoutThe destination layout

Member Function Documentation

◆ BackwardIndex()

Array<PrimExpr> tvm::tir::BijectiveLayout::BackwardIndex ( const Array< PrimExpr > &  dst_index) const

◆ BackwardShape()

Array<PrimExpr> tvm::tir::BijectiveLayout::BackwardShape ( const Array< PrimExpr > &  dst_shape) const

◆ ForwardIndex()

Array<PrimExpr> tvm::tir::BijectiveLayout::ForwardIndex ( const Array< PrimExpr > &  index) const

◆ ForwardShape()

Array<PrimExpr> tvm::tir::BijectiveLayout::ForwardShape ( const Array< PrimExpr > &  shape) const

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::tir::BijectiveLayout::TVM_DEFINE_OBJECT_REF_METHODS ( BijectiveLayout  ,
ObjectRef  ,
BijectiveLayoutNode   
)

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