diff options
| author | Diego Lopez Recas <Diego.LopezRecas@arm.com> | 2021-03-19 12:40:16 +0000 |
|---|---|---|
| committer | Colm Donelan <colm.donelan@arm.com> | 2021-04-21 14:51:17 +0000 |
| commit | fe95d7203ba7a6ac31e0656190ba56de6fcf4735 (patch) | |
| tree | 4b127a99c97364d6b9062d93ece26ccafa60579d /CMakeLists.txt | |
| parent | 7612bd6cc385dfbf54f831a6349f3a9363c6d0a2 (diff) | |
| download | armnn-fe95d7203ba7a6ac31e0656190ba56de6fcf4735.tar.gz | |
Fold PAD into Pooling2d if possible
Some models would add a PAD layer before a pooling when they can't
express their padding configuration as SAME or VALID.
ArmNN can potentially merge the two merge the two because pooling
layers are described with explicit padding.
The merge is possible if the extra padding is neutral in the combined
pooling operation. A merged operation can only fuse paddings in the
dimensions that accept explicit padding in a pooling operation, i.e. the
spatial dimensions.
Signed-off-by: Diego Lopez Recas <diego.lopez.recas@gmail.com>
Signed-off-by: Colm Donelan <Colm.Donelan@arm.com>
Change-Id: Icd54718dcd9e797c923456b7fa6e0213e288e668
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index c02db3d014..df34d28eaa 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -461,7 +461,7 @@ list(APPEND armnn_sources src/armnn/optimizations/ConvertConstants.hpp src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp - src/armnn/optimizations/FoldPadIntoConvolution2d.hpp + src/armnn/optimizations/FoldPadIntoLayer2d.hpp src/armnn/optimizations/MovePermuteUp.hpp src/armnn/optimizations/MoveTransposeUp.hpp src/armnn/optimizations/Optimization.hpp |