diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 15:11:30 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-01-23 17:27:08 +0000 |
commit | f2e175c99612189d640cb3dbb05a5dbb3c92b22f (patch) | |
tree | 618b1468af59e554181436845f4d66cc83b5373a | |
parent | d00ad916b72a53eb2488d82899ec7f033294d959 (diff) | |
download | android-nn-driver-f2e175c99612189d640cb3dbb05a5dbb3c92b22f.tar.gz |
IVGCVSW-4266 Dump float16 input/output tensors
Change-Id: Iff91ce6857a9f96a7af65f79883c1cbe59a15dd2
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
-rw-r--r-- | Utils.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
@@ -6,6 +6,7 @@ #define LOG_TAG "ArmnnDriver" #include "Utils.hpp" +#include "Half.hpp" #include <armnnUtils/Permute.hpp> @@ -241,6 +242,11 @@ void DumpTensor(const std::string& dumpDir, dumpElementFunction = &DumpTensorElement<int32_t>; break; } + case armnn::DataType::Float16: + { + dumpElementFunction = &DumpTensorElement<armnn::Half>; + break; + } default: { dumpElementFunction = nullptr; |