From e6b0e900322a92028b2d10225cd35e5346636bc1 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Thu, 8 Jun 2023 12:22:57 +0100 Subject: Fix issue with ExecuteNetwork when running with tflite executor Signed-off-by: Narumol Prangnawarat Change-Id: Id404893c47d43dafa743f4b9524001072b426509 --- tests/ExecuteNetwork/TfliteExecutor.cpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tests') diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp index 8412750951..6455650404 100644 --- a/tests/ExecuteNetwork/TfliteExecutor.cpp +++ b/tests/ExecuteNetwork/TfliteExecutor.cpp @@ -77,6 +77,10 @@ TfLiteExecutor::TfLiteExecutor(const ExecuteNetworkParams& params, armnn::IRunti } else { + if (builder(&m_TfLiteInterpreter) != kTfLiteOk) + { + LogAndThrow("Error loading the model into the TfLiteInterpreter."); + } std::cout << "Running on TfLite without ArmNN delegate\n"; } -- cgit v1.2.1