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

Fix stride height setting in input_conv test (#16813)

Browse Source
This commit is contained in:
Maciej Kwapulinski 2023-04-14 08:53:24 +02:00 committed by GitHub
parent 67aa807892
commit 435a79a2a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/tests/functional/shared_test_classes/src/subgraph/input_conv.cpp
View File

@ -79,8 +79,16 @@ void InputConvTest::SetUp() {
auto ngPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(netPrecision);
auto params = ngraph::builder::makeParams(ngPrc, { inputShape });
auto conv0 = ngraph::builder::makeConvolution(params[0], ngPrc, { kernelShape[0], kernelShape[1] }, { stride, stride }, { 0, 0 },
{ 0, 0 }, { 1, 1 }, ngraph::op::PadType::VALID, outputChannels, true,
auto conv0 = ngraph::builder::makeConvolution(params[0],
ngPrc,
{kernelShape[0], kernelShape[1]},
{kernelShape[0] > 1 ? stride : 1, stride},
{0, 0},
{0, 0},
{1, 1},
ngraph::op::PadType::VALID,
outputChannels,
true,
generateWeights(outputChannels, kernelShape[1]));
if (addReshape) {