[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:
parent
3bf34b1166
commit
818f385398
@ -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());
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user