Fix ranks check in TSConcat transformation

itikhono · Jul 25, 2024 · c6ef20d · c6ef20d
1 parent e27c6f6
commit c6ef20d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/common/transformations/src/transformations/transpose_sinking/ts_concat.cpp b/src/common/transformations/src/transformations/transpose_sinking/ts_concat.cpp
@@ -36,7 +36,7 @@ TSConcatForward::TSConcatForward() {
             return false;
         }
 
-        if (concat_node->get_output_partial_shape(0).is_dynamic()) {
+        if (concat_node->get_output_partial_shape(0).rank().is_dynamic()) {
             return false;
         }
         // todo: support dyn rank case
@@ -83,7 +83,7 @@ TSConcatBackward::TSConcatBackward() {
         }
 
         auto concat_node = as_type_ptr<ov::op::v0::Concat>(main_node);
-        if (concat_node->get_output_partial_shape(0).is_dynamic()) {
+        if (concat_node->get_output_partial_shape(0).rank().is_dynamic()) {
             return false;
         }