[CPU Test] fix any_layout test (#21490)

This commit is contained in:
River Li authored and GitHub committed 2023-12-08 03:30:52 +00:00
1 parent 68d4ed91ed
commit 072dcea299
1 file changed
+6
@@ -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());