IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Actualize DeterministicityTest: update headers and used opset (#9335)

Browse Source
This commit is contained in:
Vitaliy Urusovskij 2021-12-21 15:31:43 +03:00 committed by GitHub
parent 2d73eb6157
commit de980b8089
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/tests/functional/inference_engine/ir_serialization/deterministicity.cpp
View File

@ -10,10 +10,9 @@
#include "ie_core.hpp" #include "ie_core.hpp"
#include <openvino/runtime/runtime.hpp> #include <openvino/runtime/runtime.hpp>
#include <ngraph/function.hpp> #include <openvino/opsets/opset8.hpp>
#include <ngraph/opsets/opset5.hpp>
using namespace ngraph; using namespace ov;
#ifndef IR_SERIALIZATION_MODELS_PATH // should be already defined by cmake #ifndef IR_SERIALIZATION_MODELS_PATH // should be already defined by cmake
# error "IR_SERIALIZATION_MODELS_PATH is not defined" # error "IR_SERIALIZATION_MODELS_PATH is not defined"
@ -114,22 +113,22 @@ protected:
TEST_F(DeterministicityTest, IRInputsOrder) { TEST_F(DeterministicityTest, IRInputsOrder) {
const std::vector<std::string> friend_names = {"A", "B", "C"}; const std::vector<std::string> friend_names = {"A", "B", "C"};
auto a = std::make_shared<opset5::Parameter>(element::f32, Shape{2, 2}); auto a = std::make_shared<opset8::Parameter>(element::f32, Shape{2, 2});
auto b = std::make_shared<opset5::Parameter>(element::f32, Shape{2, 2}); auto b = std::make_shared<opset8::Parameter>(element::f32, Shape{2, 2});
auto add = std::make_shared<opset5::Add>(a, b); auto add = std::make_shared<opset8::Add>(a, b);
auto c = std::make_shared<opset5::Parameter>(element::f32, Shape{2, 2}); auto c = std::make_shared<opset8::Parameter>(element::f32, Shape{2, 2});
auto relu = std::make_shared<opset5::Relu>(c); auto relu = std::make_shared<opset8::Relu>(c);
auto add2 = std::make_shared<opset5::Add>(add, relu); auto add2 = std::make_shared<opset8::Add>(add, relu);
auto res = std::make_shared<opset5::Result>(add2); auto res = std::make_shared<opset8::Result>(add2);
a->set_friendly_name(friend_names[0]); a->set_friendly_name(friend_names[0]);
b->set_friendly_name(friend_names[1]); b->set_friendly_name(friend_names[1]);
c->set_friendly_name(friend_names[2]); c->set_friendly_name(friend_names[2]);
auto model = std::make_shared<ov::Model>(NodeVector{res}, ParameterVector{a, b, c}); auto model = std::make_shared<Model>(NodeVector{res}, ParameterVector{a, b, c});
ov::runtime::Core core; runtime::Core core;
auto serialize = ov::pass::Serialize(m_out_xml_path, m_out_bin_path); auto serialize = pass::Serialize(m_out_xml_path, m_out_bin_path);
serialize.run_on_function(model); serialize.run_on_function(model);
auto serialized_func = core.read_model(m_out_xml_path); auto serialized_func = core.read_model(m_out_xml_path);
@ -143,23 +142,23 @@ TEST_F(DeterministicityTest, IRInputsOrder) {
TEST_F(DeterministicityTest, IROutputsOrder) { TEST_F(DeterministicityTest, IROutputsOrder) {
const std::vector<std::string> friend_names = {"D", "E", "F"}; const std::vector<std::string> friend_names = {"D", "E", "F"};
auto a = std::make_shared<opset5::Parameter>(element::f32, Shape{4, 4}); auto a = std::make_shared<opset8::Parameter>(element::f32, Shape{4, 4});
auto axis_1 = ngraph::opset5::Constant::create(element::i64, Shape{}, {1}); auto axis_1 = opset8::Constant::create(element::i64, Shape{}, {1});
auto split1 = std::make_shared<opset5::Split>(a, axis_1, 2); auto split1 = std::make_shared<opset8::Split>(a, axis_1, 2);
auto res1 = std::make_shared<opset5::Result>(split1->output(0)); auto res1 = std::make_shared<opset8::Result>(split1->output(0));
auto relu = std::make_shared<opset5::Relu>(split1->output(1)); auto relu = std::make_shared<opset8::Relu>(split1->output(1));
auto split2 = std::make_shared<opset5::Split>(relu, axis_1, 2); auto split2 = std::make_shared<opset8::Split>(relu, axis_1, 2);
auto res2 = std::make_shared<opset5::Result>(split2->output(0)); auto res2 = std::make_shared<opset8::Result>(split2->output(0));
auto res3 = std::make_shared<opset5::Result>(split2->output(1)); auto res3 = std::make_shared<opset8::Result>(split2->output(1));
res1->set_friendly_name(friend_names[0]); res1->set_friendly_name(friend_names[0]);
res2->set_friendly_name(friend_names[1]); res2->set_friendly_name(friend_names[1]);
res3->set_friendly_name(friend_names[2]); res3->set_friendly_name(friend_names[2]);
auto model = std::make_shared<ov::Model>(NodeVector{res1, res2, res3}, ParameterVector{a}); auto model = std::make_shared<Model>(NodeVector{res1, res2, res3}, ParameterVector{a});
ov::runtime::Core core; runtime::Core core;
auto serialize = ov::pass::Serialize(m_out_xml_path, m_out_bin_path); auto serialize = pass::Serialize(m_out_xml_path, m_out_bin_path);
serialize.run_on_function(model); serialize.run_on_function(model);
auto serialized_func = core.read_model(m_out_xml_path); auto serialized_func = core.read_model(m_out_xml_path);