1//===- TosaOptionalDecompositions.cpp -------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// Pass to apply the Tosa operations decompositions
10// exposed as populate functions in
11// include/mlir/Dialect/Tosa/Transforms/Passes.h
12//
13//===----------------------------------------------------------------------===//
14
15#include "mlir/Dialect/Tosa/Transforms/Passes.h"
16
17#include "mlir/Dialect/Func/IR/FuncOps.h"
18#include "mlir/Dialect/Tosa/IR/TosaOps.h"
19#include "mlir/Pass/Pass.h"
20#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
21
22namespace mlir {
23namespace tosa {
24#define GEN_PASS_DEF_TOSAOPTIONALDECOMPOSITIONSPASS
25#include "mlir/Dialect/Tosa/Transforms/Passes.h.inc"
26} // namespace tosa
27} // namespace mlir
28
29using namespace mlir;
30
31namespace {
32
33struct TosaOptionalDecompositions
34 : public tosa::impl::TosaOptionalDecompositionsPassBase<
35 TosaOptionalDecompositions> {
36 void runOnOperation() override {
37 auto *ctx = &getContext();
38 RewritePatternSet patterns(ctx);
39 auto func = getOperation();
40
41 mlir::tosa::populateTosaDecomposeTransposeConv(ctx: ctx, patterns);
42 mlir::tosa::populateTosaDecomposeDepthwise(ctx: ctx, patterns);
43
44 if (applyPatternsGreedily(func, std::move(patterns)).failed())
45 signalPassFailure();
46 }
47};
48
49} // namespace
50

Provided by KDAB

Privacy Policy
Update your C++ knowledge – Modern C++11/14/17 Training
Find out more

source code of mlir/lib/Dialect/Tosa/Transforms/TosaOptionalDecompositions.cpp