VectorDropLeadUnitDim.cpp File Reference

#include <numeric>
#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h"
#include "mlir/Dialect/Vector/Transforms/VectorTransforms.h"
#include "mlir/Dialect/Vector/Utils/VectorUtils.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/TypeUtilities.h"

Go to the source code of this file.

Macros
#define	DEBUG_TYPE   "vector-drop-unit-dim"

Functions
static VectorType	trimLeadingOneDims (VectorType oldType)
static SmallVector< int64_t >	splatZero (int64_t rank)
	Return a smallVector of size rank containing all zeros. More...

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "vector-drop-unit-dim"

Definition at line 19 of file VectorDropLeadUnitDim.cpp.

Function Documentation

splatZero()

static SmallVector<int64_t> splatZero ( int64_t  rank )

static

Return a smallVector of size rank containing all zeros.

Definition at line 48 of file VectorDropLeadUnitDim.cpp.

Referenced by mlir::vector::castAwayContractionLeadingOneDim().

trimLeadingOneDims()

static VectorType trimLeadingOneDims ( VectorType  oldType )

static

Definition at line 26 of file VectorDropLeadUnitDim.cpp.

References mlir::get().