MLIR  20.0.0git
Namespaces | Macros | Functions
NamedOpConversions.cpp File Reference
#include "mlir/Dialect/Linalg/Passes.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/TypeSwitch.h"
#include "mlir/Dialect/Linalg/Passes.h.inc"

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 

Macros

#define GEN_PASS_DEF_LINALGNAMEDOPCONVERSIONPASS
 

Functions

static llvm::SmallVector< int64_t > getIndicesVector (int start, int end)
 
static LogicalResult matchAndReplaceDepthwiseConv (Operation *operation, Value input, Value kernel, Value iZp, Value kZp, Value init, Attribute stride, Attribute dilation, PatternRewriter &rewriter)
 

Macro Definition Documentation

◆ GEN_PASS_DEF_LINALGNAMEDOPCONVERSIONPASS

#define GEN_PASS_DEF_LINALGNAMEDOPCONVERSIONPASS

Definition at line 24 of file NamedOpConversions.cpp.

Function Documentation

◆ getIndicesVector()

static llvm::SmallVector<int64_t> getIndicesVector ( int  start,
int  end 
)
static

Definition at line 31 of file NamedOpConversions.cpp.

Referenced by matchAndReplaceDepthwiseConv().

◆ matchAndReplaceDepthwiseConv()

static LogicalResult matchAndReplaceDepthwiseConv ( Operation operation,
Value  input,
Value  kernel,
Value  iZp,
Value  kZp,
Value  init,
Attribute  stride,
Attribute  dilation,
PatternRewriter rewriter 
)
static