Merge pull request opencv#16089 from dkurt:dnn_ie_fix_fpga

chrisnicollo · Dec 9, 2019 · 45f6931 · 45f6931
2 parents b9435b9 + beb5b29
commit 45f6931
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/modules/dnn/src/op_inf_engine.cpp b/modules/dnn/src/op_inf_engine.cpp
@@ -750,11 +750,16 @@ void InfEngineBackendNet::initPlugin(InferenceEngine::CNNNetwork& net)
             {
                 if (layer->type == kOpenCVLayersType)
                 {
-                    layer->affinity = "CPU";
                     isHetero = true;
+#if INF_ENGINE_VER_MAJOR_LT(INF_ENGINE_RELEASE_2019R3)
+                    // Not sure about lower versions but in 2019R3 we do not need this
+                    layer->affinity = "CPU";
                 }
                 else
+                {
                     layer->affinity = device_name;
+#endif
+                }
             }
         }
         if (isHetero)