commit f2e175c99612189d640cb3dbb05a5dbb3c92b22f
author Jim Flynn <jim.flynn@arm.com> Thu Dec 12 15:11:30 2019 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Jan 23 17:27:08 2020 +0000
tree 618b1468af59e554181436845f4d66cc83b5373a
parent d00ad916b72a53eb2488d82899ec7f033294d959

IVGCVSW-4266 Dump float16 input/output tensors

Change-Id: Iff91ce6857a9f96a7af65f79883c1cbe59a15dd2
Signed-off-by: Jim Flynn <jim.flynn@arm.com>

Utils.cpp

diff --git a/Utils.cpp b/Utils.cpp
index c01604c..6ee025e 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -6,6 +6,7 @@
 #define LOG_TAG "ArmnnDriver"
 
 #include "Utils.hpp"
+#include "Half.hpp"
 
 #include <armnnUtils/Permute.hpp>
 
@@ -241,6 +242,11 @@
             dumpElementFunction = &DumpTensorElement<int32_t>;
             break;
         }
+        case armnn::DataType::Float16:
+        {
+            dumpElementFunction = &DumpTensorElement<armnn::Half>;
+            break;
+        }
         default:
         {
             dumpElementFunction = nullptr;