IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Fixed invalid vector element access in reduce test (#9538)

Browse Source
This commit is contained in:
Vladimir Paramuzov 2022-01-10 15:08:11 +03:00 committed by GitHub
parent 00361b7617
commit 8d8ceeb5d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tests/functional/shared_test_classes/src/subgraph/reduce_eltwise.cpp
View File

@ -56,7 +56,8 @@ void ReduceEltwiseTest::SetUp() {
auto reduce = std::make_shared<ngraph::opset3::ReduceSum>(paramOuts[0], reductionAxesNode, keepDims);
std::vector<size_t> constShape(reduce.get()->get_output_size(), 1);
std::vector<size_t> 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<float>(ngPrc, constShape, {}, true);
auto eltw = ngraph::builder::makeEltwise(reduce, constant, ngraph::helpers::EltwiseTypes::MULTIPLY);