From 2d9956162dd002a41f7fb4fa6753195d33524c7f Mon Sep 17 00:00:00 2001
From: David Monahan <David.Monahan@arm.com>
Date: Mon, 1 Nov 2021 10:16:37 +0000
Subject: IVGCVSW-6359 Added support for Float16 (Half) to Execute Network

 * Allows the user to specify float16 as a datatype
 * Does not contain support for float16 on the TfLiteDelegate via
   ExecuteNetwork

Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: Icba56feedab32662e2cf671cc46ada899cf40c6c
---
 .../NetworkExecutionUtils.cpp                      | 29 ++++++++++++++++++++++
 1 file changed, 29 insertions(+)

(limited to 'tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp')

diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp
index 6c74aaa6ed..00ed55caaf 100644
--- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp
+++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp
@@ -33,6 +33,15 @@ auto ParseDataArray<armnn::DataType::Float32>(std::istream& stream)
     return ParseArrayImpl<float>(stream, [](const std::string& s) { return std::stof(s); });
 }
 
+template<>
+auto ParseDataArray<armnn::DataType::Float16>(std::istream& stream)
+{
+    return ParseArrayImpl<armnn::Half>(stream, [](const std::string& s)
+    {
+        return armnn::Half(std::stof(s));
+    });
+}
+
 template<>
 auto ParseDataArray<armnn::DataType::Signed32>(std::istream& stream)
 {
@@ -139,6 +148,20 @@ void TensorPrinter::operator()(const std::vector<float>& values)
     WriteToFile(values);
 }
 
+void TensorPrinter::operator()(const std::vector<armnn::Half>& values)
+{
+    if (m_PrintToConsole)
+    {
+        std::cout << m_OutputBinding << ": ";
+        ForEachValue(values, [](armnn::Half value)
+        {
+            printf("%f ", static_cast<float>(value));
+        });
+        printf("\n");
+    }
+    WriteToFile(values);
+}
+
 void TensorPrinter::operator()(const std::vector<uint8_t>& values)
 {
     if(m_DequantizeOutput)
@@ -261,6 +284,12 @@ void PopulateTensorWithData(armnnUtils::TContainer& tensorData,
                          GenerateDummyTensorData<armnn::DataType::Float32>(numElements);
         }
     }
+    else if (dataTypeStr.compare("float16") == 0)
+    {
+        tensorData = readFromFile ?
+                     ParseDataArray<armnn::DataType::Float16>(inputTensorFile) :
+                     GenerateDummyTensorData<armnn::DataType::Float16>(numElements);
+    }
     else if (dataTypeStr.compare("int") == 0)
     {
         tensorData = readFromFile ?
-- 
cgit v1.2.1