diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-20 12:17:00 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-20 12:17:00 +0000 |
commit | 19a1c0347709506c5ad5c1b7f647a5af4a30fba0 (patch) | |
tree | 95c7fe5979ade833023b44d9198976b3915e9c02 /tests | |
parent | dc032fca290deb39af65050c254a701596b53fa8 (diff) | |
download | armnn-19a1c0347709506c5ad5c1b7f647a5af4a30fba0.tar.gz |
MLCE-336 'TfLite parser and ExecuteNetwork issues'
* Report the error thrown
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: If5e5160f43c11a284b919d76b04d9c4d53e9c56c
Diffstat (limited to 'tests')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 6880a17c52..5df5dfbce7 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -61,6 +61,11 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, armnnDelegate::TfLiteArmnnDelegateDelete); // Register armnn_delegate to TfLiteInterpreter int status = tfLiteInterpreter->ModifyGraphWithDelegate(std::move(theArmnnDelegate)); + if (status == kTfLiteError) + { + ARMNN_LOG(fatal) << "Could not register ArmNN TfLite Delegate to TfLiteInterpreter!"; + return EXIT_FAILURE; + } std::vector<std::string> inputBindings; for (const std::string& inputName: params.m_InputNames) |