[Pass][EliminateConcatStridedSlice] Add issue fix for pass EliminateConcatStridedSlice (#29555)

### Details:
Make EliminateConcatStridedSlice transformation more strict, add an
additional concat axis check.
Fixed the compile_model regression on GPU.

### Tickets:
 - CVS-164273

Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com>

Author: zhaixuejun1993

src/common/transformations/src/transformations/common_optimizations/nop_elimination.cpp

@@ -555,6 +555,24 @@ pass::EliminateConcatStridedSlice::EliminateConcatStridedSlice() {
                     return false;
                 }
 
+                // check that concatenated and split axis is the same
+                auto check_axis = [concat_axis](const std::vector<int64_t>& masks) {
+                    for (size_t axis = 0; axis < masks.size(); ++axis) {
+                        if (masks[axis] != 1 && axis != static_cast<size_t>(concat_axis)) {
+                            return false;
+                        }
+                        if (masks[axis] != 0 && axis == static_cast<size_t>(concat_axis)) {
+                            return false;
+                        }
+                    }
+                    return true;
+                };
+                auto begin_mask = strided_slice_node->get_begin_mask();
+                auto end_mask = strided_slice_node->get_end_mask();
+                if (!check_axis(begin_mask) || !check_axis(end_mask)) {
+                    return false;
+                }
+
                 auto begin_node = strided_slice_node->get_input_node_shared_ptr(1);
                 const auto& begin_constant_node = ov::util::get_constant_from_source(begin_node);
                 if (begin_constant_node == nullptr)

src/common/transformations/tests/common_optimizations/nop_elimination.cpp

@@ -1641,6 +1641,50 @@ TEST_F(TransformationTestsF, EliminateConcatStridedSliceConcat) {
         model_ref = std::make_shared<ov::Model>(ResultVector{result}, ParameterVector{param1, param2, param3});
     }
 }
+
+TEST_F(TransformationTestsF, EliminateConcatStridedSliceConcatDiffAxis) {
+    {
+        int64_t axis = 2;
+        auto param1 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 3});
+        auto param2 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 4});
+        auto param3 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 5});
+        auto concat = make_shared<ov::op::v0::Concat>(ov::as_output_vector({param1, param2, param3}), axis);
+
+        auto begin_const1 = ov::op::v0::Constant::create(ov::element::i64, ov::Shape{3}, {0, 0, 0});
+        auto end_const1 = ov::op::v0::Constant::create(ov::element::i64, ov::Shape{3}, {0, 3, 0});
+        auto strided_slice1 = std::make_shared<ov::op::v1::StridedSlice>(concat,
+                                                                         begin_const1,
+                                                                         end_const1,
+                                                                         std::vector<int64_t>{1, 0, 1},
+                                                                         std::vector<int64_t>{1, 0, 1});
+        auto relu = std::make_shared<op::v0::Relu>(strided_slice1);
+
+        auto result = std::make_shared<op::v0::Result>(relu);
+        model = std::make_shared<ov::Model>(ResultVector{result}, ParameterVector{param1, param2, param3});
+        manager.register_pass<ov::pass::EliminateConcatStridedSlice>();
+    }
+    {
+        int64_t axis = 2;
+        auto param1 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 3});
+        auto param2 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 4});
+        auto param3 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 5});
+        auto concat = make_shared<ov::op::v0::Concat>(ov::as_output_vector({param1, param2, param3}), axis);
+
+        auto begin_const1 = ov::op::v0::Constant::create(ov::element::i64, ov::Shape{3}, {0, 0, 0});
+        auto end_const1 = ov::op::v0::Constant::create(ov::element::i64, ov::Shape{3}, {0, 3, 0});
+        auto strided_slice1 = std::make_shared<ov::op::v1::StridedSlice>(concat,
+                                                                         begin_const1,
+                                                                         end_const1,
+                                                                         std::vector<int64_t>{1, 0, 1},
+                                                                         std::vector<int64_t>{1, 0, 1});
+        auto relu = std::make_shared<op::v0::Relu>(strided_slice1);
+
+        auto result = std::make_shared<op::v0::Result>(relu);
+
+        model_ref = std::make_shared<ov::Model>(ResultVector{result}, ParameterVector{param1, param2, param3});
+    }
+}
+
 TEST_F(TransformationTestsF, EliminateStridedSlice) {
     {
         auto input = std::make_shared<op::v0::Parameter>(ov::element::f32,