From a3e5b69cc961d2cab9e227c64e2362ac95b98084 Mon Sep 17 00:00:00 2001
From: Jim Flynn <jim.flynn@arm.com>
Date: Thu, 12 Dec 2019 15:11:30 +0000
Subject: IVGCVSW-4266 Dump float16 input/output tensors

Change-Id: Iff91ce6857a9f96a7af65f79883c1cbe59a15dd2
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
---
 Utils.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/Utils.cpp b/Utils.cpp
index 2f9a4a34..8af12a32 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -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;
-- 
cgit v1.2.1