From 8d8ceeb5d7ee801213590db3d00ee1c148152313 Mon Sep 17 00:00:00 2001 From: Vladimir Paramuzov Date: Mon, 10 Jan 2022 15:08:11 +0300 Subject: [PATCH] [GPU] Fixed invalid vector element access in reduce test (#9538) --- .../shared_test_classes/src/subgraph/reduce_eltwise.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp b/src/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp index 14c2cf95ff6..9778b16ec47 100644 --- a/src/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp +++ b/src/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp @@ -56,7 +56,8 @@ void ReduceEltwiseTest::SetUp() { auto reduce = std::make_shared(paramOuts[0], reductionAxesNode, keepDims); - std::vector constShape(reduce.get()->get_output_size(), 1); + std::vector constShape(reduce.get()->get_output_partial_shape(0).rank().get_length(), 1); + ASSERT_GT(constShape.size(), 2); constShape[2] = inputShape.back(); auto constant = ngraph::builder::makeConstant(ngPrc, constShape, {}, true); auto eltw = ngraph::builder::makeEltwise(reduce, constant, ngraph::helpers::EltwiseTypes::MULTIPLY);