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/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/Utils/VectorUtils.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.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
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::tensor

Macros
#define	GEN_PASS_DEF_FOLDTENSORSUBSETOPSPASS

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_FOLDTENSORSUBSETOPSPASS

#define GEN_PASS_DEF_FOLDTENSORSUBSETOPSPASS

Definition at line 28 of file FoldTensorSubsetOps.cpp.

Function Documentation

getTensorOperand() [1/2]

Value getTensorOperand ( tensor::InsertSliceOp op )

static

Definition at line 39 of file FoldTensorSubsetOps.cpp.

getTensorOperand() [2/2]

Value getTensorOperand ( vector::TransferReadOp op )

static

Definition at line 35 of file FoldTensorSubsetOps.cpp.

preconditionsFoldExtractOrInsertWithTransferOp()

template<typename XferOp, typename ExtractOrInsertOp>

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

static

Definition at line 76 of file FoldTensorSubsetOps.cpp.

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