mirror of
https://github.com/opencv/opencv.git
synced 2024-11-24 19:20:28 +08:00
Fix HETERO:FPGA,CPU plugin for IE backend
This commit is contained in:
parent
c790779a37
commit
beb5b291b9
@ -750,11 +750,16 @@ void InfEngineBackendNet::initPlugin(InferenceEngine::CNNNetwork& net)
|
|||||||
{
|
{
|
||||||
if (layer->type == kOpenCVLayersType)
|
if (layer->type == kOpenCVLayersType)
|
||||||
{
|
{
|
||||||
layer->affinity = "CPU";
|
|
||||||
isHetero = true;
|
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
|
else
|
||||||
|
{
|
||||||
layer->affinity = device_name;
|
layer->affinity = device_name;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (isHetero)
|
if (isHetero)
|
||||||
|
Loading…
Reference in New Issue
Block a user