From bc7ffb5e9e5f4c86280b20c65772eb12d8bb140e Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 20 Mar 2020 15:01:01 +0000 Subject: IVGCVSW-4520 Implement BFloat16 Optimizer * Add ReduceFp32ToBf16 to OptimizerOptions * Add ConvertFp32NetworkToBf16 * Add utility functions to insert conversion layers * Add constant conversion BF16 <-> FP32 * Unit tests Signed-off-by: Narumol Prangnawarat Change-Id: Iaff77e20c721400b052cb37eb9ef6fe16d7abaff --- src/armnn/optimizations/All.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnn/optimizations/All.hpp') diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index 273c337665..9fc284213d 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -6,6 +6,7 @@ #include "AddDebug.hpp" #include "ConvertConstants.hpp" +#include "ConvertFp32NetworkToBf16.hpp" #include "ConvertFp32NetworkToFp16.hpp" #include "FoldPadIntoConvolution2d.hpp" #include "MovePermuteUp.hpp" -- cgit v1.2.1