From 2a9eec1c3fbfd1f4c3b90c531e3d4d94d28d8f42 Mon Sep 17 00:00:00 2001
From: Gleb Kazantaev <gleb.kazantaev@intel.com>
Date: Tue, 17 Aug 2021 14:39:05 +0300
Subject: [PATCH] Enable SplitSqueezeConcatFusion + TransposeSinking in offline
 transformations (#7087)

---
 .../src/moc_transformations.cpp                        | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/inference-engine/src/offline_transformations/src/moc_transformations.cpp b/inference-engine/src/offline_transformations/src/moc_transformations.cpp
index 745e173b4d7..d12801d2ce8 100644
--- a/inference-engine/src/offline_transformations/src/moc_transformations.cpp
+++ b/inference-engine/src/offline_transformations/src/moc_transformations.cpp
@@ -8,9 +8,19 @@
 #include "pruning.hpp"
 
 #include <ngraph/pass/manager.hpp>
+#include <transformations/common_optimizations/split_squeeze_concat_fusion.hpp>
+#include <transformations/common_optimizations/transpose_sinking.hpp>
 
 NGRAPH_RTTI_DEFINITION(ngraph::pass::MOCTransformations, "MOCTransformations", 0);
 
 bool ngraph::pass::MOCTransformations::run_on_function(std::shared_ptr<ngraph::Function> f) {
+    ngraph::pass::Manager manager(get_pass_config());
+    auto transpose_sinking = manager.register_pass<ngraph::pass::GraphRewrite>();
+    transpose_sinking->add_matcher<ngraph::pass::TransposeSinking>();
+    // SplitSqueezeConcatFusion should work in same GraphRewrite as TransposesSinking,
+    // because it replaces pattern that may contain Transposes which must be optimized before
+    // the transformation and it also inserts Transpose that can be optimized by TransposeSinking
+    transpose_sinking->add_matcher<ngraph::pass::SplitSqueezeConcatFusion>();
+    manager.run_passes(f);
     return false;
 }
\ No newline at end of file