Using IE backend for normalize layer tests

This commit is contained in:
Liubov Batanina 2019-02-07 11:52:27 +03:00
parent f67b197d49
commit b068d26fad

View File

@ -65,16 +65,12 @@ public:
{ {
if (backendId == DNN_BACKEND_INFERENCE_ENGINE) if (backendId == DNN_BACKEND_INFERENCE_ENGINE)
{ {
if (pnorm != 2) if (!(pnorm == 2 || pnorm == 1)) {
return false; return false;
if (!blobs.empty()) }
return true; return preferableTarget == DNN_TARGET_MYRIAD ? !acrossSpatial : startAxis == 1;
if (preferableTarget == DNN_TARGET_MYRIAD)
return !acrossSpatial;
return startAxis == 1 && (!acrossSpatial || endAxis > 1);
} }
else return backendId == DNN_BACKEND_OPENCV;
return backendId == DNN_BACKEND_OPENCV;
} }
bool getMemoryShapes(const std::vector<MatShape> &inputs, bool getMemoryShapes(const std::vector<MatShape> &inputs,