Revert "Add SegmentMax-16 reference implementation" (#29037)

Reverts #28788
Merged by mistake in queue settings, there are
[failures](https://openvino-ci.toolbox.iotg.sclab.intel.com/job/private-ci/job/ie/job/build-windows-vs2022/6458/)
on Windows

Author: akladiev

src/core/include/openvino/op/util/attr_types.hpp

@@ -20,7 +20,7 @@ enum class PadMode { CONSTANT = 0, EDGE, REFLECT, SYMMETRIC };
 OPENVINO_API
 std::ostream& operator<<(std::ostream& s, const PadMode& type);
 
-/// \brief Fill modes to set default value for operators like `SegmentMax`.
+/// \brief Fill modes for the `SegmentMax` operator.
 enum class FillMode { ZERO = 0, LOWEST };
 
 OPENVINO_API

src/core/include/openvino/opsets/opset16_tbl.hpp

@@ -16,4 +16,3 @@ _OPENVINO_OP_REG(ShapeOf, ov::op::v3)
 // New operations added in opset16
 _OPENVINO_OP_REG(Identity, ov::op::v16)
 _OPENVINO_OP_REG(ISTFT, ov::op::v16)
-_OPENVINO_OP_REG(SegmentMax, ov::op::v16)

src/core/reference/include/openvino/reference/segment_max.hpp

@@ -50,9 +50,8 @@ std::vector<TRShape> shape_infer(const SegmentMax* op,
 
     // validate num_segments input
     const auto num_segments_available = op->inputs().size() == 3;
-    const auto& num_segments = num_segments_available
-                                   ? ov::op::get_input_const_data_as<TRShape, int64_t>(op, 2, tensor_accessor)
-                                   : ov::optional<std::vector<int64_t>>{};
+    const auto num_segments = num_segments_available ? get_input_const_data_as_shape<TRShape>(op, 2, tensor_accessor)
+                                                     : ov::optional<TRShape>{};
     if (num_segments_available) {
         const auto& num_segments_shape = input_shapes[2];
         NODE_SHAPE_INFER_CHECK(op,

src/core/shape_inference/include/segment_max_shape_inference.hpp

@@ -77,7 +77,7 @@ INSTANTIATE_TEST_SUITE_P(opset,
                                          OpsetTestParams{ov::get_opset13, 186},
                                          OpsetTestParams{ov::get_opset14, 188},
                                          OpsetTestParams{ov::get_opset15, 199},
-                                         OpsetTestParams{ov::get_opset16, 6}),
+                                         OpsetTestParams{ov::get_opset16, 5}),
                          OpsetTestNameGenerator{});
 
 class MyOpOld : public ov::op::Op {

src/core/tests/opset.cpp

@@ -16,7 +16,6 @@
 
 namespace ov::test {
 using op::v0::Constant, op::v0::Parameter, op::v1::Add, op::v1::ReduceMax, op::v1::StridedSlice, op::v3::ShapeOf;
-using testing::HasSubstr;
 
 class TypePropSegmentMaxTest : public TypePropOpTest<op::v16::SegmentMax> {};
 
@@ -70,44 +69,45 @@ TEST_F(TypePropSegmentMaxTest, incorrect_inputs) {
         const auto num_segments_f32 = std::make_shared<Parameter>(element::f32, PartialShape{});
         OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids, num_segments_f32, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("The element type of the num_segments input be i32 or i64."));
+                        testing::HasSubstr("The element type of the num_segments input be i32 or i64."));
     }
     {
         const auto segment_ids_f32 = std::make_shared<Parameter>(element::f32, PartialShape{3});
         OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids_f32, num_segments, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("The element type of the segment_ids input be i32 or i64."));
+                        testing::HasSubstr("The element type of the segment_ids input be i32 or i64."));
     }
     {
         const auto segment_ids_nd = std::make_shared<Parameter>(element::i32, PartialShape{2, 3});
         OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids_nd, num_segments, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("segment_ids must be a 1D input."));
+                        testing::HasSubstr("segment_ids must be a 1D input."));
     }
     {
         const auto num_segments_nd = std::make_shared<Parameter>(element::i32, PartialShape{1});
         OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids, num_segments_nd, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("num_segments must be a scalar input."));
+                        testing::HasSubstr("num_segments must be a scalar input."));
     }
     {
         const auto segment_ids_unsorted =
             std::make_shared<Constant>(element::i32, Shape{3}, std::vector<int64_t>{1, 0, 1});
         OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids_unsorted, num_segments, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("segment_ids must be sorted."));
+                        testing::HasSubstr("segment_ids must be sorted."));
     }
     {
         const auto data_scalar = std::make_shared<Parameter>(element::i32, PartialShape{});
         OV_EXPECT_THROW(std::ignore = make_op(data_scalar, segment_ids, num_segments, op::FillMode::LOWEST),
                         ov::NodeValidationFailure,
-                        HasSubstr("The data input cannot be a scalar."));
+                        testing::HasSubstr("The data input cannot be a scalar."));
     }
     {
         const auto segment_ids_short = std::make_shared<Constant>(element::i32, Shape{2}, std::vector<int64_t>{1, 0});
-        OV_EXPECT_THROW(std::ignore = make_op(data, segment_ids_short, num_segments, op::FillMode::LOWEST),
-                        ov::NodeValidationFailure,
-                        HasSubstr("The number of elements in segment_ids must match the first dimension of data."));
+        OV_EXPECT_THROW(
+            std::ignore = make_op(data, segment_ids_short, num_segments, op::FillMode::LOWEST),
+            ov::NodeValidationFailure,
+            testing::HasSubstr("The number of elements in segment_ids must match the first dimension of data."));
     }
 }
 

src/core/tests/type_prop/segment_max.cpp

@@ -558,7 +558,3 @@ extern template bool evaluate_node<ov::op::v15::SearchSorted>(std::shared_ptr<ov
 extern template bool evaluate_node<ov::op::v16::Identity>(std::shared_ptr<ov::Node> node,
                                                           ov::TensorVector& outputs,
                                                           const ov::TensorVector& inputs);
-
-extern template bool evaluate_node<ov::op::v16::SegmentMax>(std::shared_ptr<ov::Node> node,
-                                                            ov::TensorVector& outputs,
-                                                            const ov::TensorVector& inputs);

src/plugins/template/backend/ops/ops_evaluates.hpp

@@ -177,7 +177,6 @@ _OPENVINO_OP_REG(SearchSorted, ov::op::v15)
 
 _OPENVINO_OP_REG(Identity, ov::op::v16)
 _OPENVINO_OP_REG(ISTFT, ov::op::v16)
-_OPENVINO_OP_REG(SegmentMax, ov::op::v16)
 
 _OPENVINO_OP_REG(AUGRUCell, ov::op::internal)
 _OPENVINO_OP_REG(AUGRUSequence, ov::op::internal)