commit 19a1c0347709506c5ad5c1b7f647a5af4a30fba0
author Sadik Armagan <sadik.armagan@arm.com> Wed Jan 20 12:17:00 2021 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Wed Jan 20 12:17:00 2021 +0000
tree 95c7fe5979ade833023b44d9198976b3915e9c02
parent dc032fca290deb39af65050c254a701596b53fa8

MLCE-336 'TfLite parser and ExecuteNetwork issues'

* Report the error thrown

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: If5e5160f43c11a284b919d76b04d9c4d53e9c56c

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index 8e0fae6..a18471a 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -2130,6 +2130,10 @@
             // Extract target shape from input
             auto bufferPtr = GetBuffer(m_Model, inputs[1]->buffer);
             auto values = reinterpret_cast<const int32_t*>(bufferPtr->data.data());
+            if (!values)
+            {
+                ARMNN_THROW_PARSE_EXCEPTION("Reshape operator target shape input buffer data is null");
+            }
             for (int i=0; i < inputs[1]->shape[0]; ++i)
             {
                 targetShape.push_back(values[i]);

tests/ExecuteNetwork/ExecuteNetwork.cpp

diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 6880a17..5df5dfb 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -61,6 +61,11 @@
                              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)