diff options
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 6 | ||||
-rw-r--r-- | tests/ExecuteNetwork/TfliteExecutor.cpp | 5 |
2 files changed, 7 insertions, 4 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index c6c8cc0b27..a70f33ff23 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -20,7 +20,7 @@ std::unique_ptr<IExecutor> BuildExecutor(ProgramOptions& programOptions) return std::make_unique<TfLiteExecutor>(programOptions.m_ExNetParams); #else ARMNN_LOG(fatal) << "Not built with Arm NN Tensorflow-Lite delegate support."; - return nullptr; + return nullptr; #endif } else @@ -59,7 +59,7 @@ int main(int argc, const char* argv[]) try { executor = BuildExecutor(programOptions); - if (executor->m_constructionFailed) + if ((!executor) || (executor->m_constructionFailed)) { return EXIT_FAILURE; } 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; |