Fixed header paths for some nGraph ops

* Added dependency on IE version

backport of commit: 992c908b56
This commit is contained in:
Ilya Churaev
2020-07-06 11:11:59 +03:00
committed by Alexander Alekhin
parent 65dbbf712d
commit d69a7a3bbf
6 changed files with 28 additions and 1 deletions

View File

@@ -48,9 +48,14 @@
#ifdef HAVE_DNN_NGRAPH
#include "../ie_ngraph.hpp"
#if INF_ENGINE_VER_MAJOR_GT(INF_ENGINE_RELEASE_2020_4)
#include <ngraph/op/roi_pooling.hpp>
#include <ngraph/op/psroi_pooling.hpp>
#else
#include <ngraph/op/experimental/layers/roi_pooling.hpp>
#include <ngraph/op/experimental/layers/psroi_pooling.hpp>
#endif
#endif
#include <float.h>
#include <algorithm>