MLIR
17.0.0git
|
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/Utils/StaticValueUtils.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/Support/Debug.h"
Go to the source code of this file.
Functions | |
static Value | createInserts (RewriterBase &rewriter, Location loc, int dim, Value destination, ArrayRef< int64_t > shape, ArrayRef< Value > constants, OperandRange::iterator &elementIt, SmallVectorImpl< Value > &indices) |
static Operation * | movePaddingToFillOrGenericOp (RewriterBase &rewriter, Location loc, PadOp padOp, Value dest) |
static SmallVector< Value > | reifyOrComputeDynamicSizes (OpBuilder &b, Value value) |
static Value | createAllocationForTensor (RewriterBase &rewriter, Location loc, Value value, Attribute memorySpace={}) |
|
static |
Definition at line 152 of file ConvertToDestinationStyle.cpp.
Referenced by mlir::linalg::bufferizeToAllocation().
|
static |
Definition at line 34 of file ConvertToDestinationStyle.cpp.
References mlir::OpBuilder::create().
Referenced by mlir::linalg::rewriteInDestinationPassingStyle().
|
static |
Definition at line 56 of file ConvertToDestinationStyle.cpp.
References mlir::OpBuilder::create(), mlir::OpBuilder::createBlock(), mlir::Builder::getContext(), mlir::Value::getDefiningOp(), mlir::MLIRContext::getLoadedDialect(), mlir::Value::getLoc(), mlir::Builder::getMultiDimIdentityMap(), mlir::Operation::getParentOp(), mlir::Value::getType(), mlir::m_Constant(), mlir::matchPattern(), mlir::Dialect::materializeConstant(), mlir::RewriterBase::mergeBlocks(), mlir::RewriterBase::replaceOpWithNewOp(), and mlir::OpBuilder::setInsertionPointToStart().
Referenced by mlir::linalg::rewriteInDestinationPassingStyle().
|
static |
Definition at line 121 of file ConvertToDestinationStyle.cpp.
References mlir::OpBuilder::create(), mlir::Value::getDefiningOp(), mlir::Value::getLoc(), mlir::Value::getType(), mlir::reifyResultShapes(), and mlir::succeeded().