Skip to content

Commit 435a79a

Browse files
authored
Fix stride height setting in input_conv test (openvinotoolkit#16813)
1 parent 67aa807 commit 435a79a

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

src/tests/functional/shared_test_classes/src/subgraph/input_conv.cpp

+11-3
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,17 @@ void InputConvTest::SetUp() {
7979
auto ngPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(netPrecision);
8080
auto params = ngraph::builder::makeParams(ngPrc, { inputShape });
8181

82-
auto conv0 = ngraph::builder::makeConvolution(params[0], ngPrc, { kernelShape[0], kernelShape[1] }, { stride, stride }, { 0, 0 },
83-
{ 0, 0 }, { 1, 1 }, ngraph::op::PadType::VALID, outputChannels, true,
84-
generateWeights(outputChannels, kernelShape[1]));
82+
auto conv0 = ngraph::builder::makeConvolution(params[0],
83+
ngPrc,
84+
{kernelShape[0], kernelShape[1]},
85+
{kernelShape[0] > 1 ? stride : 1, stride},
86+
{0, 0},
87+
{0, 0},
88+
{1, 1},
89+
ngraph::op::PadType::VALID,
90+
outputChannels,
91+
true,
92+
generateWeights(outputChannels, kernelShape[1]));
8593

8694
if (addReshape) {
8795
size_t numOutputWidth = (((inputShape[1] * inputShape[2] * inputShape[3] - kernelShape[1] * kernelShape[0]) / (inputShape[1] * stride)) + 1);

0 commit comments

Comments
 (0)