diff options
author | Diego Lopez Recas <Diego.LopezRecas@arm.com> | 2021-03-19 12:40:16 +0000 |
---|---|---|
committer | Diego Lopez Recas <diego.lopez.recas@gmail.com> | 2021-03-22 10:35:04 +0000 |
commit | 51ce7d487c761358de105f82ff90553570aedac0 (patch) | |
tree | 763e1380249af2dee57e98b022b915c9cae8a1ee /CMakeLists.txt | |
parent | 8b35d0922e040ecf4d6b2f5012af356c383f5561 (diff) | |
download | armnn-51ce7d487c761358de105f82ff90553570aedac0.tar.gz |
Fold PAD into Pooling2d
Some models would add a PAD layer before a pooling when they can't
express their padding configuration as SAME or VALID.
Arm NN can merge the two because pooling layers are described with
explicit padding.
Signed-off-by: Diego Lopez Recas <diego.lopez.recas@gmail.com>
Change-Id: Id048186db6a005e0257bfbc1406c3b0dab2cdd58
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 4e75c28da0..25d9472413 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -483,7 +483,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 |