[luci/pass] Revise ForwardTransposeOpPass (Samsung#13224)

This will revise ForwardTransposeOpPass to check I/O rank are same. ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
mhs4670go · Jun 18, 2024 · c9799f1 · c9799f1
1 parent 61b116f
commit c9799f1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/luci/pass/src/ForwardTransposeOpPass.cpp b/compiler/luci/pass/src/ForwardTransposeOpPass.cpp
@@ -164,6 +164,7 @@ class EBOWithConstPattern final : public CircleNodeMutableVisitor<bool>
     {
       if (auto y = dynamic_cast<luci::CircleTranspose *>(node->y()))
       {
+        RETURN_FALSE_UNLESS(node->rank() == y->rank());
         RETURN_FALSE_UNLESS(check_rank_four(x));
         RETURN_FALSE_UNLESS(check_perm(y));
 
@@ -190,6 +191,7 @@ class EBOWithConstPattern final : public CircleNodeMutableVisitor<bool>
     {
       if (auto x = dynamic_cast<luci::CircleTranspose *>(node->x()))
       {
+        RETURN_FALSE_UNLESS(node->rank() == x->rank());
         RETURN_FALSE_UNLESS(check_rank_four(y));
         RETURN_FALSE_UNLESS(check_perm(x));