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 14:50:06 +0000 |
| commit | dda061addba93a91eebb9d008615f4958b09e237 (patch) | |
| tree | b4358959ce28d7c1c092efbc3761b8fcb34f4045 | |
| parent | e131a0b179babb03bf9e372a145fa464f916ff81 (diff) | |
| download | armnn-dda061addba93a91eebb9d008615f4958b09e237.tar.gz | |
IVGCVSW-4170 Also convert constants to FP16 when model converted
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I6318a70bd6818db55401bc667c116596e4909673
| -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 |