aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-04-30 11:06:34 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2020-04-30 10:23:34 +0000
commitb248ec1f47467d835b1baea1f5a3a726f8e585e2 (patch)
treeba6faa9c8f7eff2c62eb23b50167ec5790f9455c
parentd46368765369a8058398e01361cd22a7793894dd (diff)
downloadandroid-nn-driver-b248ec1f47467d835b1baea1f5a3a726f8e585e2.tar.gz
IVGCVSW-4771 Add support for Boolean and QAsymmS8 in dump files
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I89c245c0d3ca291e1d9864121b68762205278f35
-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;