aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--Utils.cpp10
1 files changed, 10 insertions, 0 deletions
diff --git a/Utils.cpp b/Utils.cpp
index a7df499c..6481c287 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -326,6 +326,16 @@ void DumpTensor(const std::string& dumpDir,
dumpElementFunction = &DumpTensorElement<armnn::Half>;
break;
}
+ case armnn::DataType::QAsymmS8:
+ {
+ dumpElementFunction = &DumpTensorElement<int8_t, int32_t>;
+ break;
+ }
+ case armnn::DataType::Boolean:
+ {
+ dumpElementFunction = &DumpTensorElement<bool>;
+ break;
+ }
default:
{
dumpElementFunction = nullptr;