[ShapeInference] GridSample shape infer review (#15102)

* Add more type_prop tests for interval dims and labels

* Add setter for grid sample attributes

* Merge grid sample batch dims

* Add StaticShapeInferenceTest for GridSample

* Fix label test

* Use OpStaticShapeInferenceTest fixture in test

src/plugins/intel_cpu/tests/unit/shape_inference_test/grid_sample_shape_inference_test.cpp

@@ -0,0 +1,39 @@
+// Copyright (C) 2018-2023 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+//
+
+#include <gtest/gtest.h>
+
+#include "utils.hpp"
+#include "openvino/opsets/opset9.hpp"
+#include "grid_sample_shape_inference.hpp"
+
+using namespace ov;
+using namespace ov::intel_cpu;
+
+class GridSampleStaticShapeInferenceTest : public OpStaticShapeInferenceTest<opset9::GridSample> {};
+
+TEST_F(GridSampleStaticShapeInferenceTest, GridSample) {
+    const auto data = std::make_shared<opset9::Parameter>(element::i32, PartialShape{-1, -1, -1, -1});
+    const auto grid = std::make_shared<opset9::Parameter>(element::f32, PartialShape{-1, -1, -1, -1});
+
+    op = make_op(data, grid, opset9::GridSample::Attributes{});
+
+    input_shapes = {StaticShape{2, 3, 4, 8}, StaticShape{2, 6, 7, 2}};
+    output_shapes = {StaticShape{}};
+    exp_shape = StaticShape{2, 3, 6, 7};
+
+    shape_inference(op.get(), input_shapes, output_shapes);
+    EXPECT_EQ(output_shapes[0], exp_shape);
+}
+
+TEST_F(GridSampleStaticShapeInferenceTest, GridSample_default_constructor) {
+    op = make_op();
+
+    input_shapes = {StaticShape{2, 3, 4, 8}, StaticShape{2, 6, 7, 2}};
+    output_shapes = {StaticShape{}};
+    exp_shape = StaticShape{2, 3, 6, 7};
+
+    shape_infer(op.get(), input_shapes, output_shapes);
+    EXPECT_EQ(output_shapes[0], exp_shape);
+}