MLIR  20.0.0git
Macros | Functions
VectorDropLeadUnitDim.cpp File Reference
#include <numeric>
#include "mlir/Dialect/Arith/IR/Arith.h"
#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 20 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 49 of file VectorDropLeadUnitDim.cpp.

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

◆ trimLeadingOneDims()

static VectorType trimLeadingOneDims ( VectorType  oldType)
static

Definition at line 27 of file VectorDropLeadUnitDim.cpp.

References mlir::get().