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

FEM test (#7899)

Browse Source
This commit is contained in:
Ilya Lavrenov 2021-10-14 12:38:42 +03:00 committed by GitHub
parent 21fba21667
commit d2a671b3cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ngraph/test/frontend/shared/src/load_from.cpp
View File

@ -6,6 +6,7 @@
#include <fstream>
#include "openvino/core/variant.hpp"
#include "utils.hpp"
using namespace ngraph;
@ -40,6 +41,25 @@ TEST_P(FrontEndLoadFromTest, testLoadFromFilePath) {
ASSERT_NE(function, nullptr);
}
TEST_P(FrontEndLoadFromTest, testLoadFromFilePathWithExplicitVariants) {
std::string model_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_file);
std::vector<std::string> frontends;
FrontEnd::Ptr fe;
ASSERT_NO_THROW(frontends = m_fem.get_available_front_ends());
std::vector<std::shared_ptr<ov::Variant>> variants;
variants.emplace_back(ov::make_variant(model_path));
ASSERT_NO_THROW(m_frontEnd = m_fem.load_by_model(variants));
ASSERT_NE(m_frontEnd, nullptr);
ASSERT_NO_THROW(m_inputModel = m_frontEnd->load(variants));
ASSERT_NE(m_inputModel, nullptr);
std::shared_ptr<ov::Function> function;
ASSERT_NO_THROW(function = m_frontEnd->convert(m_inputModel));
ASSERT_NE(function, nullptr);
}
TEST_P(FrontEndLoadFromTest, testLoadFromTwoFiles) {
std::string model_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_files[0]);
std::string weights_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_files[1]);