Add functionRefs for PrePostProcessTest

2021-09-24 14:44:23 +09:00 · 2021-09-24 14:44:23 +09:00 · ab476a9ba3
commit ab476a9ba3
parent 451ce9096d
1 changed files with 1 additions and 0 deletions
--- a/inference-engine/tests/functional/shared_test_classes/src/subgraph/preprocess.cpp
+++ b/inference-engine/tests/functional/shared_test_classes/src/subgraph/preprocess.cpp
@ -31,6 +31,7 @@ void PrePostProcessTest::SetUp() {
    function = (std::get<0>(func))();
    outPrc = InferenceEngine::details::convertPrecision(function->get_output_element_type(0));
    threshold = std::get<2>(func);
+    functionRefs = ngraph::clone_function(*function);
 }

 TEST_P(PrePostProcessTest, CompareWithRefs) {