From 51ce7d487c761358de105f82ff90553570aedac0 Mon Sep 17 00:00:00 2001 From: Diego Lopez Recas Date: Fri, 19 Mar 2021 12:40:16 +0000 Subject: 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 Change-Id: Id048186db6a005e0257bfbc1406c3b0dab2cdd58 --- src/armnn/optimizations/All.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnn/optimizations/All.hpp') diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index d042616ba4..5decc7c969 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -9,7 +9,7 @@ #include "ConvertConstants.hpp" #include "ConvertFp32NetworkToBf16.hpp" #include "ConvertFp32NetworkToFp16.hpp" -#include "FoldPadIntoConvolution2d.hpp" +#include "FoldPadIntoLayer2d.hpp" #include "FuseBatchNorm.hpp" #include "MovePermuteUp.hpp" #include "MoveTransposeUp.hpp" -- cgit v1.2.1