diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2022-07-22 10:23:41 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-07-27 15:58:02 +0100 |
commit | 4073e147982df3022ea97cb85179c8c7a695a17a (patch) | |
tree | 657123d837744e982cc87c4ef024a5fbb60a0c8a /src/armnn/optimizations/All.hpp | |
parent | 1e276f38e67af7505a25010eee579034ee83d12b (diff) | |
download | armnn-4073e147982df3022ea97cb85179c8c7a695a17a.tar.gz |
IVGCVSW-6978: RedirectMembersToConstantInputs does not work with Fp32NetworkToBf16Converter
* Fuse FP32ToBF16Layers with Constant Layer so Conv2d/FullyConnected
can have their weights redirected.
* If BF16 Unsupported in Conv2d || FullyConnected revert fused
Constant Layer to FP32
Change-Id: If523c708a822659d64597d9ae39cca1c2f84b76f
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/armnn/optimizations/All.hpp')
-rw-r--r-- | src/armnn/optimizations/All.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index 900e763762..0421f31973 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -9,6 +9,7 @@ #include "ConvertConstants.hpp" #include "ConvertConstDequantisationLayersToConstLayers.hpp" #include "ConvertConstPermuteLayersToConstLayers.hpp" +#include "FuseConvertFp32ToBf16IntoConstLayers.hpp" #include "ConvertFp32NetworkToBf16.hpp" #include "ConvertFp32NetworkToFp16.hpp" #include "FoldPadIntoLayer2d.hpp" |