[luci/pass] Introduce RemoveUnnecessaryTransposeNetPass (Samsung#11976)

This PR introduces RemoveUnnecessaryTransposeNetPass and related unit tests. ONE-DCO-1.0-Signed-off-by: SeungHui Youn <[email protected]>
mhs4670go · Nov 10, 2023 · c492527 · c492527
1 parent ee6e037
commit c492527
Show file tree

Hide file tree

Showing 5 changed files with 755 additions and 0 deletions.
diff --git a/compiler/luci/pass/include/luci/CircleOptimizer.h b/compiler/luci/pass/include/luci/CircleOptimizer.h
@@ -84,6 +84,7 @@ class CircleOptimizer final
       RemoveUnnecessaryStridedSlice,
       RemoveUnnecessarySplit,
       RemoveUnnecessaryReshape,
+      RemoveUnnecessaryTranspose,
       TransformMinMaxToRelu6Pass,
       TransformMinReluToRelu6Pass,
       DecomposeHardSwishPass,

diff --git a/compiler/luci/pass/include/luci/Pass/RemoveUnnecessaryTransposeNetPass.h b/compiler/luci/pass/include/luci/Pass/RemoveUnnecessaryTransposeNetPass.h
@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef __LUCI_REMOVE_UNNECESSARY_TRANSPOSE_NET_PASS_H__
+#define __LUCI_REMOVE_UNNECESSARY_TRANSPOSE_NET_PASS_H__
+
+#include <logo/Pass.h>
+
+namespace luci
+{
+
+struct RemoveUnnecessaryTransposeNetPass final : public logo::Pass
+{
+  const char *name(void) const final { return "luci::RemoveUnnecessaryTransposeNetPass"; }
+
+  bool run(loco::Graph *g) final;
+};
+
+} // namespace luci
+
+#endif // __LUCI_REMOVE_UNNECESSARY_TRANSPOSE_NET_PASS_H__
diff --git a/compiler/luci/pass/src/CircleOptimizer.cpp b/compiler/luci/pass/src/CircleOptimizer.cpp
@@ -57,6 +57,7 @@
 #include "luci/Pass/RemoveUnnecessarySlicePass.h"
 #include "luci/Pass/RemoveUnnecessaryStridedSlicePass.h"
 #include "luci/Pass/RemoveUnnecessarySplitPass.h"
+#include "luci/Pass/RemoveUnnecessaryTransposeNetPass.h"
 #include "luci/Pass/ReplaceNonConstFCWithBatchMatMulPass.h"
 #include "luci/Pass/ReplaceMulAddWithDepthwiseConvPass.h"
 #include "luci/Pass/ReplaceSubWithAddPass.h"
@@ -401,6 +402,10 @@ void CircleOptimizer::optimize(loco::Graph *g) const
   {
     phase.emplace_back(std::make_unique<luci::RemoveUnnecessarySplitPass>());
   }
+  if (_options->query(Options::Algorithm::RemoveUnnecessaryTranspose))
+  {
+    phase.emplace_back(std::make_unique<luci::RemoveUnnecessaryTransposeNetPass>());
+  }
   if (_options->query(Options::Algorithm::RemoveRedundantReshape))
   {
     phase.emplace_back(std::make_unique<luci::RemoveRedundantReshapePass>());