diff --git a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/any_layout.cpp b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/any_layout.cpp index 84984dda2bf..c3770469921 100644 --- a/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/any_layout.cpp +++ b/src/plugins/intel_cpu/tests/functional/subgraph_tests/src/any_layout.cpp @@ -3,6 +3,7 @@ // #include "test_utils/cpu_test_utils.hpp" +#include "openvino/core/preprocess/pre_post_process.hpp" namespace ov { namespace test { @@ -41,6 +42,11 @@ protected: // Create model auto function = create_test_function(shape); + auto ppp_model = ov::preprocess::PrePostProcessor(function); + ppp_model.input().tensor().set_layout("..."); + ppp_model.output().tensor().set_layout("..."); + function = ppp_model.build(); + auto input = ov::Tensor(ov::element::f32, shape, input_data.data()); auto output = ov::Tensor(ov::element::f32, shape, output_data.data());