diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-08-16 12:17:24 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-29 10:12:44 +0100 |
commit | ca565c1b04b767abfc13a59146680663a3ea4008 (patch) | |
tree | 58b260768fe8cb690f740ece44fca9c415620d7f /delegate/include | |
parent | 514d16b434102a4a7807548745af301baab13b6b (diff) | |
download | armnn-ca565c1b04b767abfc13a59146680663a3ea4008.tar.gz |
IVGCVSW-6603 'Add a no fallback mode to the TfLite Delegate'
* Added disable-tflite-runtime-fallback option to armnn_delegate
* Updated armnn_delegate version
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I449b16404d3ffe98e6dac52a43e7c25225addd73
Diffstat (limited to 'delegate/include')
-rw-r--r-- | delegate/include/DelegateOptions.hpp | 18 | ||||
-rw-r--r-- | delegate/include/Version.hpp | 4 |
2 files changed, 20 insertions, 2 deletions
diff --git a/delegate/include/DelegateOptions.hpp b/delegate/include/DelegateOptions.hpp index d789ea7285..2b0107e834 100644 --- a/delegate/include/DelegateOptions.hpp +++ b/delegate/include/DelegateOptions.hpp @@ -174,6 +174,12 @@ public: * This is an Experimental parameter that is incompatible with "infer-output-shape". \n * This parameter may be removed in a later update. * + * Option key: "disable-tflite-runtime-fallback" \n + * Possible values: ["true"/"false"] \n + * Description: Disable TfLite Runtime fallback in the Arm NN TfLite delegate. + * An exception will be thrown if unsupported operators are encountered. + * This option is only for testing purposes. + * * @param[in] option_keys Delegate option names * @param[in] options_values Delegate option values * @param[in] num_options Number of delegate options @@ -262,6 +268,15 @@ public: return m_RuntimeOptions; } + void DisableTfLiteRuntimeFallback(bool fallbackState) + { + m_DisableTfLiteRuntimeFallback = fallbackState; + } + bool TfLiteRuntimeFallbackDisabled() + { + return m_DisableTfLiteRuntimeFallback; + } + private: /// Which backend to run Delegate on. /// Examples of possible values are: CpuRef, CpuAcc, GpuAcc. @@ -295,6 +310,9 @@ private: /// If not empty then the optimized model will be serialized to a file with this file name in "dot" format. std::string m_SerializeToDot = ""; + + /// Option to disable TfLite Runtime fallback for unsupported operators. + bool m_DisableTfLiteRuntimeFallback = false; }; } // namespace armnnDelegate diff --git a/delegate/include/Version.hpp b/delegate/include/Version.hpp index c14857e320..36d8fb54af 100644 --- a/delegate/include/Version.hpp +++ b/delegate/include/Version.hpp @@ -13,8 +13,8 @@ namespace armnnDelegate #define STRINGIFY_MACRO(s) #s // ArmNN Delegate version components -#define DELEGATE_MAJOR_VERSION 26 -#define DELEGATE_MINOR_VERSION 1 +#define DELEGATE_MAJOR_VERSION 27 +#define DELEGATE_MINOR_VERSION 0 #define DELEGATE_PATCH_VERSION 0 /// DELEGATE_VERSION: "X.Y.Z" |