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>
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).
◆ BijectiveLayout()
tvm::tir::BijectiveLayout::BijectiveLayout |
( |
Layout |
src_layout, |
|
|
Layout |
dst_layout |
|
) |
| |
The constructor.
- Parameters
-
src_layout | The source layout |
dst_layout | The destination layout |
◆ 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()
The documentation for this class was generated from the following file: