From 19a1c0347709506c5ad5c1b7f647a5af4a30fba0 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 20 Jan 2021 12:17:00 +0000 Subject: MLCE-336 'TfLite parser and ExecuteNetwork issues' * Report the error thrown Signed-off-by: Sadik Armagan Change-Id: If5e5160f43c11a284b919d76b04d9c4d53e9c56c --- src/armnnTfLiteParser/TfLiteParser.cpp | 4 ++++ tests/ExecuteNetwork/ExecuteNetwork.cpp | 5 +++++ 2 files changed, 9 insertions(+) diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 8e0fae68d1..a18471a71e 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -2130,6 +2130,10 @@ void TfLiteParser::ParseReshape(size_t subgraphIndex, size_t operatorIndex) // Extract target shape from input auto bufferPtr = GetBuffer(m_Model, inputs[1]->buffer); auto values = reinterpret_cast(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]); 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 inputBindings; for (const std::string& inputName: params.m_InputNames) -- cgit v1.2.1