From dd6804bdbc6a1213fd8b636fad1af80c775faf23 Mon Sep 17 00:00:00 2001
From: Derek Lamberti <derek.lamberti@arm.com>
Date: Wed, 27 Nov 2019 09:29:57 +0000
Subject: IVGCVSW-4170 Also convert constants to FP16 when model converted

Change-Id: Ideeec890a91379bfd52774633b1b35fa1b14f434
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
---
 src/armnn/Network.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index d16475947d..53e28c344a 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -842,6 +842,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork,
     if (options.m_ReduceFp32ToFp16)
     {
         Optimizer::Pass(optGraph, MakeOptimizations(Fp32NetworkToFp16Converter()));
+        Optimizer::Pass(optGraph, MakeOptimizations(ConvertConstantsFloatToHalf()));
     }
 
     // Initialize backend settings
-- 
cgit v1.2.1