aboutsummaryrefslogtreecommitdiff
path: root/tests/ExecuteNetwork/TfliteExecutor.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/ExecuteNetwork/TfliteExecutor.cpp')
-rw-r--r--tests/ExecuteNetwork/TfliteExecutor.cpp9
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");