MLIR
15.0.0git
|
#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h"
#include "mlir/Dialect/Vector/Utils/VectorUtils.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/ImplicitLocOpBuilder.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... | |
#define DEBUG_TYPE "vector-drop-unit-dim" |
Definition at line 17 of file VectorDropLeadUnitDim.cpp.
|
static |
Return a smallVector of size rank
containing all zeros.
Definition at line 35 of file VectorDropLeadUnitDim.cpp.
References mlir::Type::cast(), mlir::OpBuilder::create(), mlir::Type::dyn_cast(), mlir::detail::enumerate(), mlir::failure(), mlir::AffineMap::get(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getAffineMapArrayAttr(), mlir::Builder::getArrayAttr(), mlir::Operation::getAttrs(), mlir::Builder::getContext(), mlir::Builder::getI64Type(), mlir::OperationName::getIdentifier(), mlir::Operation::getLoc(), mlir::Operation::getName(), mlir::AffineMap::getNumDims(), mlir::Operation::getNumResults(), mlir::AffineMap::getNumSymbols(), mlir::Operation::getOperands(), mlir::AffineMap::getPermutationMap(), mlir::Operation::getResult(), mlir::AffineMap::getResults(), mlir::Operation::getResultTypes(), mlir::Builder::getZeroAttr(), mlir::OpTrait::hasElementwiseMappableTraits(), mlir::isParallelIterator(), mlir::OpRewritePattern< SourceOp >::OpRewritePattern(), perm(), mlir::RewriterBase::replaceOpWithNewOp(), mlir::success(), mlir::transform::detail::transposeResults(), and trimLeadingOneDims().
|
static |
Definition at line 24 of file VectorDropLeadUnitDim.cpp.
Referenced by splatZero().