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
---
 tests/InferenceTest.inl | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'tests/InferenceTest.inl')

diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl
index b6087c5e5a..94dbfe78b8 100644
--- a/tests/InferenceTest.inl
+++ b/tests/InferenceTest.inl
@@ -67,6 +67,14 @@ struct ClassifierResultProcessor
                                 });
     }
 
+    void operator()(const std::vector<armnn::Half>& values)
+    {
+        SortPredictions(values, [](armnn::Half value)
+                                {
+                                    return value;
+                                });
+    }
+
     void operator()(const std::vector<int8_t>& values)
     {
         SortPredictions(values, [](int8_t value)
-- 
cgit v1.2.1