change inputShape

andrei-cv · Jun 7, 2021 · 1ea081c · 1ea081c
1 parent 03415e5
commit 1ea081c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp b/inference-engine/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp
@@ -57,7 +57,7 @@ void ReduceEltwiseTest::SetUp() {
 
     auto reduce = std::make_shared<ngraph::opset3::ReduceSum>(paramOuts[0], reductionAxesNode, keepDims);
 
-    std::vector<size_t> constShape(3, 1);
+    std::vector<size_t> constShape(reduce.get()->get_output_size(), 1);
     constShape[2] = inputShape.back();
     auto constant = ngraph::builder::makeConstant<float>(ngPrc, constShape, {}, true);
     auto eltw = ngraph::builder::makeEltwise(reduce, constant, ngraph::helpers::EltwiseTypes::MULTIPLY);