commit 39831da6a91ae808b5c5b904add2c75304fbacf7
author Ryan OShea <ryan.oshea3@arm.com> Thu Jan 26 17:43:45 2023 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Tue Jan 31 16:20:42 2023 +0000
tree 0345f618faefead4cc81ec05480bac6a2655df78
parent 1cebf4978bf7723aaf0501de5fb80a6ef77066bf

Fix for ExecuteNetwork Delegate when output is boolean

 * When the output of a network is a boolean from a comparison layer
   ExecuteNetwork was missing the data type when writing the output tensor

Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Id9c1609462395a68e8c1842c77a4a033a10f74e8

tests/ExecuteNetwork/TfliteExecutor.cpp

diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp
index 3c8313b..f01abfc 100644
--- a/tests/ExecuteNetwork/TfliteExecutor.cpp
+++ b/tests/ExecuteNetwork/TfliteExecutor.cpp
@@ -200,6 +200,15 @@
                         }
                         break;
                     }
+                    case kTfLiteBool:
+                    {
+                        auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<bool>(
+                                tfLiteDelegateOutputId);
+                        for (int i = 0; i < outputSize; ++i) {
+                            fprintf(outputTensorFile, "%u ", tfLiteDelegateOutputData[i]);
+                        }
+                        break;
+                    }
                     default:
                     {
                         LogAndThrow("Unsupported output type");