diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-15 14:07:05 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-05-18 15:31:51 +0000 |
commit | 07307f3c40c4efd8615755ed92ce300a3e150732 (patch) | |
tree | fef7e7068a613e34491a7c846dda9b19b61e3a8f /src/armnn/optimizations/All.hpp | |
parent | 80512b0e1e98cb6782f5526fc8308ae6a571cd59 (diff) | |
download | armnn-07307f3c40c4efd8615755ed92ce300a3e150732.tar.gz |
IVGCVSW-6455 Support Const + Dequantize layer and optimize it.
* Support Float16 as input to Dequantize layer
* Add Optimization to substitute Const+Dequantize layers with Const layer
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I58bb7e3871ca480c7b6fca93c4efb2de84e09e64
Signed-off-by: David <david.monahan@arm.com>
Diffstat (limited to 'src/armnn/optimizations/All.hpp')
-rw-r--r-- | src/armnn/optimizations/All.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index 38c4ac9462..e4a1f33e08 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -7,6 +7,7 @@ #include "AddBroadcastReshapeLayer.hpp" #include "AddDebug.hpp" #include "ConvertConstants.hpp" +#include "ConvertConstDequantisationLayersToConstLayers.hpp" #include "ConvertFp32NetworkToBf16.hpp" #include "ConvertFp32NetworkToFp16.hpp" #include "FoldPadIntoLayer2d.hpp" |