diff options
author | Colm Donelan <colm.donelan@arm.com> | 2023-01-24 22:10:12 +0000 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2023-01-25 11:28:42 +0000 |
commit | 18e6f04232b8aecaec9a232057ac3cb16e4f56f5 (patch) | |
tree | 87144020143756a5ff496eccca6f79075b5e29bb /tests/ExecuteNetwork/TfliteExecutor.cpp | |
parent | 04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f (diff) | |
download | armnn-18e6f04232b8aecaec9a232057ac3cb16e4f56f5.tar.gz |
IVGCVSW-7441 Fixing null pointers in ExecuteNetwork.
* Check if BuildExecutor returns null in ExecuteNetwork.
* Check if tflite BuildFromFile returns null in TfliteExecutor.
Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I42b6e5f26dfd127dd16b6b322184900846317c41
Diffstat (limited to 'tests/ExecuteNetwork/TfliteExecutor.cpp')
-rw-r--r-- | tests/ExecuteNetwork/TfliteExecutor.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp index 6a799548bf..810495fe8c 100644 --- a/tests/ExecuteNetwork/TfliteExecutor.cpp +++ b/tests/ExecuteNetwork/TfliteExecutor.cpp @@ -8,7 +8,10 @@ TfLiteExecutor::TfLiteExecutor(const ExecuteNetworkParams& params) : m_Params(params) { m_Model = tflite::FlatBufferModel::BuildFromFile(m_Params.m_ModelPath.c_str()); - + if (!m_Model) + { + LogAndThrow("Failed to load TfLite model from: " + m_Params.m_ModelPath); + } m_TfLiteInterpreter = std::make_unique<Interpreter>(); tflite::ops::builtin::BuiltinOpResolver resolver; |