diff options
Diffstat (limited to 'delegate/opaque/include/armnn_delegate.hpp')
-rw-r--r-- | delegate/opaque/include/armnn_delegate.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/delegate/opaque/include/armnn_delegate.hpp b/delegate/opaque/include/armnn_delegate.hpp index e08f466f09..653015a63c 100644 --- a/delegate/opaque/include/armnn_delegate.hpp +++ b/delegate/opaque/include/armnn_delegate.hpp @@ -7,9 +7,14 @@ #include <DelegateOptions.hpp> +#include <tensorflow/core/public/version.h> #include <tensorflow/lite/c/c_api_opaque.h> #include <tensorflow/lite/core/experimental/acceleration/configuration/c/stable_delegate.h> +#if TF_MAJOR_VERSION > 2 || (TF_MAJOR_VERSION == 2 && TF_MINOR_VERSION > 5) +#define ARMNN_POST_TFLITE_2_5 +#endif + namespace armnnOpaqueDelegate { |