diff --git a/src/common/transformations/src/transformations/transpose_sinking/ts_concat.cpp b/src/common/transformations/src/transformations/transpose_sinking/ts_concat.cpp index 8dbcf7ba285f5b..93a3bc0a67a881 100644 --- 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(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; }