[IE TESTS] Add example for QueryNetwork in SLT (#7628)

* [IE TESTS] Add example for QueryNetwork in SLT

* Update mul_conv_fusion.cpp

* Skip

* skip[
This commit is contained in:
Irina Efode authored and GitHub committed 2021-09-27 11:22:38 +03:00
1 parent 3bf34b1166
commit 818f385398
5 files changed
+38 -1

No files matched your search

@@ -50,8 +50,10 @@ std::vector<std::string> disabledTestPatterns() {
R"(.*smoke_MemoryTest.*iteration_count=3.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*iteration_count=4.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*iteration_count=10.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*LOW_LATENCY.*iteration_count=10.*IS=\(1.2\).*)",
R"(.*smoke_MemoryTest.*LOW_LATENCY.*iteration_count=10.*IS=\(1.2\).*)",/**/
// CVS-58963: Not implemented yet
R"(.*Behavior.*InferRequest.*OutOfFirstOutIsInputForSecondNetwork.*)",
// TODO: Issue: 29577
R"(.*QueryNetwork.*)",
};
}
@@ -20,4 +20,16 @@ TEST_P(ActivationDynamicLayerTest, CompareWithRefs) {
Run();
}
TEST_P(ActivationLayerTest, QueryNetwork) {
QueryNetwork();
}
TEST_P(ActivationParamLayerTest, QueryNetwork) {
QueryNetwork();
}
TEST_P(ActivationDynamicLayerTest, QueryNetwork) {
QueryNetwork();
}
} // namespace LayerTestsDefinitions
@@ -12,4 +12,8 @@ TEST_P(SplitConvConcat, CompareWithRefImpl) {
Run();
};
TEST_P(SplitConvConcat, QueryNetwork) {
QueryNetwork();
}
} // namespace SubgraphTestsDefinitions
@@ -57,6 +57,8 @@ public:
virtual void Serialize();
virtual void QueryNetwork();
static void Compare(const std::vector<std::pair<ngraph::element::Type, std::vector<std::uint8_t>>> &expected,
const std::vector<InferenceEngine::Blob::Ptr> &actual,
float threshold);
@@ -90,6 +90,23 @@ void LayerTestsCommon::Serialize() {
CommonTestUtils::removeIRFiles(out_xml_path, out_bin_path);
}
void LayerTestsCommon::QueryNetwork() {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
cnnNetwork = InferenceEngine::CNNNetwork(function);
auto queryNetworkResult = PluginCache::get().ie()->QueryNetwork(cnnNetwork, targetDevice);
std::set<std::string> expected;
for (auto&& node : function->get_ops()) {
expected.insert(node->get_friendly_name());
}
std::set<std::string> actual;
for (auto&& res : queryNetworkResult.supportedLayersMap) {
actual.insert(res.first);
}
ASSERT_EQ(expected, actual);
}
InferenceEngine::Blob::Ptr LayerTestsCommon::GenerateInput(const InferenceEngine::InputInfo &info) const {
return FuncTestUtils::createAndFillBlob(info.getTensorDesc());
}