LowerVectorTransfer.cpp File Reference

#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/Vector/Transforms/LoweringPatterns.h"

Go to the source code of this file.

Functions
static ArrayAttr	inverseTransposeInBoundsAttr (OpBuilder &builder, ArrayAttr attr, const SmallVector< unsigned > &permutation)
	Transpose a vector transfer op's in_bounds attribute by applying reverse permutation based on the given indices.
static Value	insertUnitDims (OpBuilder &builder, Location loc, Value vec, ArrayRef< int64_t > unitDimPositions)
	Insert unit dims into vec so that they are at unitDimPositions in the result, e.g.

Function Documentation

insertUnitDims()

Value insertUnitDims ( OpBuilder & builder, Location loc, Value vec, ArrayRef< int64_t > unitDimPositions )

static

Insert unit dims into vec so that they are at unitDimPositions in the result, e.g.

position 1 turns vector<4x8xf32> into vector<4x1x8xf32>.

Definition at line 35 of file LowerVectorTransfer.cpp.

References broadcast(), mlir::Value::getType(), and mlir::isIdentityPermutation().

inverseTransposeInBoundsAttr()

ArrayAttr inverseTransposeInBoundsAttr ( OpBuilder & builder, ArrayAttr attr, const SmallVector< unsigned > & permutation )

static

Transpose a vector transfer op's in_bounds attribute by applying reverse permutation based on the given indices.

Definition at line 23 of file LowerVectorTransfer.cpp.

References ArrayAttr(), and mlir::Builder::getBoolArrayAttr().