diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 15:11:30 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 15:11:30 +0000 |
commit | a3e5b69cc961d2cab9e227c64e2362ac95b98084 (patch) | |
tree | 011844b070dc36d64261de8fb8350e1e53f4c759 | |
parent | c7d0d44812336f4db8ab9a649fb40b4d4ed27e97 (diff) | |
download | android-nn-driver-a3e5b69cc961d2cab9e227c64e2362ac95b98084.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 | 5 |
1 files changed, 5 insertions, 0 deletions
@@ -226,6 +226,11 @@ void DumpTensor(const std::string& dumpDir, dumpElementFunction = &DumpTensorElement<int32_t>; break; } + case armnn::DataType::Float16: + { + dumpElementFunction = &DumpTensorElement<armnn::Half>; + break; + } default: { dumpElementFunction = nullptr; |