FoldTensorSubsetOps.cpp File Reference

#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Affine/ViewLikeInterfaceUtils.h"
#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/Tensor/Transforms/Passes.h"
#include "mlir/Dialect/Tensor/Transforms/Transforms.h"
#include "mlir/Dialect/Utils/IndexingUtils.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "llvm/ADT/TypeSwitch.h"
#include <type_traits>
#include "mlir/Dialect/Tensor/Transforms/Passes.h.inc"

Go to the source code of this file.

Classes
struct	InsertSliceOfInsertSliceFolder< OpTy >

Namespaces
	mlir
	Include the generated interface declarations.
	mlir::tensor

Macros
#define	GEN_PASS_DEF_FOLDTENSORSUBSETOPS

Functions
static Value	getTensorOperand (vector::TransferReadOp op)
static Value	getTensorOperand (tensor::InsertSliceOp op)
template<typename XferOp , typename ExtractOrInsertOp >
static LogicalResult	preconditionsFoldExtractOrInsertWithTransferOp (RewriterBase &rewriter, XferOp xferOp, ExtractOrInsertOp extractOrInsertSliceOp)

Macro Definition Documentation

GEN_PASS_DEF_FOLDTENSORSUBSETOPS

#define GEN_PASS_DEF_FOLDTENSORSUBSETOPS

Definition at line 29 of file FoldTensorSubsetOps.cpp.

Function Documentation

getTensorOperand() [1/2]

static Value getTensorOperand ( tensor::InsertSliceOp  op )

static

Definition at line 40 of file FoldTensorSubsetOps.cpp.

getTensorOperand() [2/2]

static Value getTensorOperand ( vector::TransferReadOp  op )

static

Definition at line 36 of file FoldTensorSubsetOps.cpp.

preconditionsFoldExtractOrInsertWithTransferOp()

template<typename XferOp , typename ExtractOrInsertOp >

static LogicalResult preconditionsFoldExtractOrInsertWithTransferOp ( RewriterBase &  rewriter, XferOp  xferOp, ExtractOrInsertOp  extractOrInsertSliceOp  )

static

Definition at line 71 of file FoldTensorSubsetOps.cpp.

References mlir::RewriterBase::notifyMatchFailure(), and mlir::success().