diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2023-01-26 17:43:45 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-01-31 16:20:42 +0000 |
commit | 39831da6a91ae808b5c5b904add2c75304fbacf7 (patch) | |
tree | 0345f618faefead4cc81ec05480bac6a2655df78 /tests/ExecuteNetwork/TfliteExecutor.cpp | |
parent | 1cebf4978bf7723aaf0501de5fb80a6ef77066bf (diff) | |
download | armnn-39831da6a91ae808b5c5b904add2c75304fbacf7.tar.gz |
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
Diffstat (limited to 'tests/ExecuteNetwork/TfliteExecutor.cpp')
-rw-r--r-- | tests/ExecuteNetwork/TfliteExecutor.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp index 3c8313b938..f01abfccd7 100644 --- a/tests/ExecuteNetwork/TfliteExecutor.cpp +++ b/tests/ExecuteNetwork/TfliteExecutor.cpp @@ -200,6 +200,15 @@ std::vector<const void *> TfLiteExecutor::Execute() } 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"); |