MLIR
20.0.0git
|
Rewrite extract_slice(tensor.pad(x)) into tensor.pad(extract_slice(x)). More...
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
Public Types | |
using | ControlFn = std::function< std::optional< bool >(tensor::ExtractSliceOp)> |
A function to control pattern application and rewrite logic. More... | |
Public Member Functions | |
ExtractSliceOfPadTensorSwapPattern (MLIRContext *context, ControlFn controlFn=nullptr, PatternBenefit benefit=1) | |
LogicalResult | matchAndRewrite (tensor::ExtractSliceOp sliceOp, PatternRewriter &rewriter) const override |
Public Member Functions inherited from mlir::OpRewritePattern< tensor::ExtractSliceOp > | |
OpRewritePattern (MLIRContext *context, PatternBenefit benefit=1, ArrayRef< StringRef > generatedNames={}) | |
Patterns must specify the root operation name they match against, and can also specify the benefit of the pattern matching and a list of generated ops. More... | |
Public Member Functions inherited from mlir::detail::OpOrInterfaceRewritePatternBase< SourceOp > | |
void | rewrite (Operation *op, PatternRewriter &rewriter) const final |
Wrappers around the RewritePattern methods that pass the derived op type. More... | |
LogicalResult | match (Operation *op) const final |
Attempt to match against code rooted at the specified operation, which is the same operation code as getRootKind(). More... | |
LogicalResult | matchAndRewrite (Operation *op, PatternRewriter &rewriter) const final |
Attempt to match against code rooted at the specified operation, which is the same operation code as getRootKind(). More... | |
virtual void | rewrite (SourceOp op, PatternRewriter &rewriter) const |
Rewrite and Match methods that operate on the SourceOp type. More... | |
virtual LogicalResult | match (SourceOp op) const |
virtual LogicalResult | matchAndRewrite (SourceOp op, PatternRewriter &rewriter) const |
Public Member Functions inherited from mlir::RewritePattern | |
virtual | ~RewritePattern ()=default |
Public Member Functions inherited from mlir::Pattern | |
ArrayRef< OperationName > | getGeneratedOps () const |
Return a list of operations that may be generated when rewriting an operation instance with this pattern. More... | |
std::optional< OperationName > | getRootKind () const |
Return the root node that this pattern matches. More... | |
std::optional< TypeID > | getRootInterfaceID () const |
Return the interface ID used to match the root operation of this pattern. More... | |
std::optional< TypeID > | getRootTraitID () const |
Return the trait ID used to match the root operation of this pattern. More... | |
PatternBenefit | getBenefit () const |
Return the benefit (the inverse of "cost") of matching this pattern. More... | |
bool | hasBoundedRewriteRecursion () const |
Returns true if this pattern is known to result in recursive application, i.e. More... | |
MLIRContext * | getContext () const |
Return the MLIRContext used to create this pattern. More... | |
StringRef | getDebugName () const |
Return a readable name for this pattern. More... | |
void | setDebugName (StringRef name) |
Set the human readable debug name used for this pattern. More... | |
ArrayRef< StringRef > | getDebugLabels () const |
Return the set of debug labels attached to this pattern. More... | |
void | addDebugLabels (ArrayRef< StringRef > labels) |
Add the provided debug labels to this pattern. More... | |
void | addDebugLabels (StringRef label) |
Additional Inherited Members | |
Static Public Member Functions inherited from mlir::RewritePattern | |
template<typename T , typename... Args> | |
static std::unique_ptr< T > | create (Args &&...args) |
This method provides a convenient interface for creating and initializing derived rewrite patterns of the given type T . More... | |
Protected Member Functions inherited from mlir::RewritePattern | |
Pattern (StringRef rootName, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Inherit the base constructors from Pattern . More... | |
Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Inherit the base constructors from Pattern . More... | |
Pattern (MatchInterfaceOpTypeTag tag, TypeID interfaceID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Inherit the base constructors from Pattern . More... | |
Pattern (MatchTraitOpTypeTag tag, TypeID traitID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Inherit the base constructors from Pattern . More... | |
Protected Member Functions inherited from mlir::Pattern | |
Pattern (StringRef rootName, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Construct a pattern with a certain benefit that matches the operation with the given root name. More... | |
Pattern (MatchAnyOpTypeTag tag, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Construct a pattern that may match any operation type. More... | |
Pattern (MatchInterfaceOpTypeTag tag, TypeID interfaceID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Construct a pattern that may match any operation that implements the interface defined by the provided interfaceID . More... | |
Pattern (MatchTraitOpTypeTag tag, TypeID traitID, PatternBenefit benefit, MLIRContext *context, ArrayRef< StringRef > generatedNames={}) | |
Construct a pattern that may match any operation that implements the trait defined by the provided traitID . More... | |
void | setHasBoundedRewriteRecursion (bool hasBoundedRecursionArg=true) |
Set the flag detailing if this pattern has bounded rewrite recursion or not. More... | |
Rewrite extract_slice(tensor.pad(x)) into tensor.pad(extract_slice(x)).
Definition at line 1598 of file Transforms.h.
using mlir::linalg::ExtractSliceOfPadTensorSwapPattern::ControlFn = std::function<std::optional<bool>(tensor::ExtractSliceOp)> |
A function to control pattern application and rewrite logic.
The function will be given the slice op and should return:
See the documentation for tensor::bubbleUpPadSlice regarding zero slice guard.
Definition at line 1609 of file Transforms.h.
|
inline |
Definition at line 1611 of file Transforms.h.
|
override |
Definition at line 991 of file Transforms.cpp.
References mlir::tensor::bubbleUpPadSlice(), and mlir::RewriterBase::replaceOp().