diff --git a/docs/template_plugin/src/template_plugin.cpp b/docs/template_plugin/src/template_plugin.cpp index 934036d3b20..393497bf040 100644 --- a/docs/template_plugin/src/template_plugin.cpp +++ b/docs/template_plugin/src/template_plugin.cpp @@ -152,14 +152,24 @@ InferenceEngine::QueryNetworkResult Plugin::QueryNetwork(const InferenceEngine:: // So we need store as supported either unsupported node sets std::unordered_set supported; std::unordered_set unsupported; - auto opset = ngraph::get_opset4(); + ngraph::OpSet op_super_set; +#define _OPENVINO_OP_REG(NAME, NAMESPACE) op_super_set.insert(); +#include "openvino/opsets/opset1_tbl.hpp" +#include "openvino/opsets/opset2_tbl.hpp" +#include "openvino/opsets/opset3_tbl.hpp" +#include "openvino/opsets/opset4_tbl.hpp" +#include "openvino/opsets/opset5_tbl.hpp" +#include "openvino/opsets/opset6_tbl.hpp" +#include "openvino/opsets/opset7_tbl.hpp" +#include "openvino/opsets/opset8_tbl.hpp" +#undef _OPENVINO_OP_REG for (auto&& node : transformedFunction->get_ops()) { // Extract transformation history from transformed node as list of nodes for (auto&& fusedLayerName : ngraph::getFusedNamesVector(node)) { // Filter just nodes from original operation set // TODO: fill with actual decision rules based on whether kernel is supported by backend if (InferenceEngine::details::contains(originalOps, fusedLayerName)) { - if (opset.contains_type(friendlyNameToType[fusedLayerName])) { + if (op_super_set.contains_type(friendlyNameToType[fusedLayerName])) { supported.emplace(fusedLayerName); } else { unsupported.emplace(fusedLayerName); diff --git a/inference-engine/tests/functional/shared_test_classes/src/base/layer_test_utils.cpp b/inference-engine/tests/functional/shared_test_classes/src/base/layer_test_utils.cpp index 7a030097121..5b0af47276b 100644 --- a/inference-engine/tests/functional/shared_test_classes/src/base/layer_test_utils.cpp +++ b/inference-engine/tests/functional/shared_test_classes/src/base/layer_test_utils.cpp @@ -96,6 +96,7 @@ void LayerTestsCommon::Serialize(ngraph::pass::Serialize::Version ir_version) { void LayerTestsCommon::QueryNetwork() { SKIP_IF_CURRENT_TEST_IS_DISABLED(); + cnnNetwork = InferenceEngine::CNNNetwork(function); auto queryNetworkResult = PluginCache::get().ie()->QueryNetwork(cnnNetwork, targetDevice);