diff --git a/src/tests/unit/cpu/shape_inference_test/scatter_elements_update_shape_inference.cpp b/src/tests/unit/cpu/shape_inference_test/scatter_elements_update_shape_inference.cpp index 491069475a2..935a1fac113 100644 --- a/src/tests/unit/cpu/shape_inference_test/scatter_elements_update_shape_inference.cpp +++ b/src/tests/unit/cpu/shape_inference_test/scatter_elements_update_shape_inference.cpp @@ -24,7 +24,7 @@ TEST(StaticShapeInferenceTest, ScatterElementsUpdateTest) { int32_t axis_shape_val[] = {2}; std::map> constant_data; constant_data[3] = - std::make_shared(ngraph::element::Type_t::i32, Shape{2}, axis_shape_val); + std::make_shared(ngraph::element::Type_t::i32, Shape{1}, axis_shape_val); std::vector input_shapes = {StaticShape{1000, 256, 7, 7}, StaticShape{125, 20, 7, 6}, StaticShape{125, 20, 7, 6},