diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2022-07-22 10:23:41 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2022-07-22 10:23:41 +0100 |
commit | 89cc6b39376419565505c59c693a107a5edd47aa (patch) | |
tree | 657123d837744e982cc87c4ef024a5fbb60a0c8a /CMakeLists.txt | |
parent | 5b1f53975d7fa482475af781ffa2a2277f40e0c9 (diff) | |
download | armnn-89cc6b39376419565505c59c693a107a5edd47aa.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 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 41db8661d3..f0eb81cc6c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -547,6 +547,7 @@ if(BUILD_UNIT_TESTS) src/armnn/test/optimizations/Fp32NetworkToFp16ConverterTests.cpp src/armnn/test/optimizations/FuseActivationTests.cpp src/armnn/test/optimizations/FuseBatchNormTests.cpp + src/armnn/test/optimizations/FuseConvertF32BF16IntoConstLayerTests.cpp src/armnn/test/optimizations/InsertDebugLayerTests.cpp src/armnn/test/optimizations/MovePermuteUpTests.cpp src/armnn/test/optimizations/MoveTransposeUpTests.cpp |