From 20ca25bfca4e307414bd7e200bbbe57b2d71b20c Mon Sep 17 00:00:00 2001
From: Sofya Balandina <sofya.balandina@intel.com>
Date: Wed, 21 Dec 2022 10:18:41 +0000
Subject: [PATCH] Fix crash on
 CompileModelCacheTestBase.CompareWithRefImpl/KSOFunction (#14678)

---
 .../include/ngraph_functions/subgraph_builders.hpp            | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/subgraph_builders.hpp b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/subgraph_builders.hpp
index 3bfa43f5edf..d563690fd9f 100644
--- a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/subgraph_builders.hpp
+++ b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/subgraph_builders.hpp
@@ -196,13 +196,13 @@ inline std::shared_ptr<ngraph::Function> makeSplitConvConcat(std::vector<size_t>
 
 inline std::shared_ptr<ngraph::Function> makeKSOFunction(std::vector<size_t> inputShape = {1, 4, 20, 20},
                                                          ngraph::element::Type_t ngPrc = ngraph::element::Type_t::f32) {
-    auto params = ngraph::builder::makeParams(ngPrc, {inputShape});
+    auto params = ngraph::builder::makeParams(ngPrc, {inputShape, inputShape});
 
     auto shapeOf = std::make_shared<ngraph::opset4::ShapeOf>(params[0]);
     auto convert = std::make_shared<ngraph::opset4::Convert>(shapeOf, ngPrc);
     auto newShape = ngraph::builder::makeConstant<int64_t>(ngraph::element::i64, {4}, {1, 4, 1, 1});
     auto reshape = std::make_shared<ngraph::opset4::Reshape>(convert, newShape, false);
-    auto conv1 = ngraph::builder::makeConvolution(params[0], ngPrc, {3, 3}, {1, 1}, {0, 0}, {0, 0}, {1, 1},
+    auto conv1 = ngraph::builder::makeConvolution(params[1], ngPrc, {3, 3}, {1, 1}, {0, 0}, {0, 0}, {1, 1},
                                                   ngraph::op::PadType::EXPLICIT, 4);
     auto relu1 = std::make_shared<ngraph::opset4::Relu>(conv1);
     auto add = std::make_shared<ngraph::opset4::Add>(relu1, reshape);