Fix HETERO:FPGA,CPU plugin for IE backend
authorDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Fri, 6 Dec 2019 16:34:54 +0000 (19:34 +0300)
committerDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Fri, 6 Dec 2019 16:35:11 +0000 (19:35 +0300)
modules/dnn/src/op_inf_engine.cpp

index 0a35194..c6f741f 100644 (file)
@@ -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)