diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-11-27 09:29:57 +0000 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2019-11-27 16:30:11 +0000 |
commit | dd6804bdbc6a1213fd8b636fad1af80c775faf23 (patch) | |
tree | 24912e31c63b03afa9ab0f8eade32e6829a71e78 /src | |
parent | 710f664e41803bcb04403a1da27990d76faab6cf (diff) | |
download | armnn-dd6804bdbc6a1213fd8b636fad1af80c775faf23.tar.gz |
IVGCVSW-4170 Also convert constants to FP16 when model converted
Change-Id: Ideeec890a91379bfd52774633b1b35fa1b14f434
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src')
-rw-r--r-- | src/armnn/Network.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
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 |