third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Track Hetero execution for nGraph networks using ngraph::Function

Browse Source
This commit is contained in:
Dmitry Kurtaev 2020-02-27 17:45:28 +03:00
parent 85c97f772c
commit 5bbe116f89
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/dnn/src/ie_ngraph.cpp
View File

@ -442,13 +442,14 @@ void InfEngineNgraphNet::initPlugin(InferenceEngine::CNNNetwork& net)
config.emplace("VPU_DETECT_NETWORK_BATCH", CONFIG_VALUE(NO));
}
bool isHetero = false;
if (device_name != "CPU")
bool isHetero = device_name == "FPGA";
// It is actual only for non-CPU targets and networks built in runtime using nGraph.
// We do not check IR models because they can be with version less than IRv10
if (!isHetero && device_name != "CPU" && !hasNetOwner)
{
isHetero = device_name == "FPGA";
for (auto& layer : net)
for (auto& node : net.getFunction()->get_ops())
{
if (layer->type == kOpenCVLayersType)
if (node->description() == kOpenCVLayersType)
{
isHetero = true;
break;