commit dd6804bdbc6a1213fd8b636fad1af80c775faf23
author Derek Lamberti <derek.lamberti@arm.com> Wed Nov 27 09:29:57 2019 +0000
committer Kevin May <kevin.may@arm.com> Wed Nov 27 16:30:11 2019 +0000
tree 24912e31c63b03afa9ab0f8eade32e6829a71e78
parent 710f664e41803bcb04403a1da27990d76faab6cf

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

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index d164759..53e28c3 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -842,6 +842,7 @@
     if (options.m_ReduceFp32ToFp16)
     {
         Optimizer::Pass(optGraph, MakeOptimizations(Fp32NetworkToFp16Converter()));
+        Optimizer::Pass(optGraph, MakeOptimizations(ConvertConstantsFloatToHalf()));
     }
 
     // Initialize backend settings