From 516272aeee5cc34598b68baa8ee3de7cfdc39f1a Mon Sep 17 00:00:00 2001 From: Vitaliy Urusovskij Date: Thu, 23 Dec 2021 13:12:45 +0300 Subject: [PATCH] Fix stack-buffer-overflow in ScatterElementsUpdateTest (#9387) --- .../scatter_elements_update_shape_inference.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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},