commit 06e0300ccf279c6b0fcbb5ef3b6fa36e00229492
author Teresa Charlin <teresa.charlinreyes@arm.com> Thu Oct 15 13:16:07 2020 +0100
committer Jim Flynn <jim.flynn@arm.com> Thu Oct 29 19:15:01 2020 +0000
tree cea4eec69904c40a326b3e4c043c88e441b77b7a
parent 34515a1897410adc08390888a6643db390a53d05

IVGCVSW-5314 Create OptimizeForExclusiveConnection

* FuseBatchNorm class has been added to facilitate testing
* Only Convolution2D FP32 being fused

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I049c4770946ddca21b08516d4c9f4d0d22bf9b45

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 373f999..6578b84 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1054,7 +1054,8 @@
                                                 OptimizeConsecutiveReshapes(),
                                                 FoldPadIntoConvolution2d(),
                                                 PermuteAndBatchToSpaceAsDepthToSpace(),
-                                                TransposeAndBatchToSpaceAsDepthToSpace()));
+                                                TransposeAndBatchToSpaceAsDepthToSpace(),
+                                                FuseBatchNormIntoConvolution2D()));
 
     // If Fp32 to Fp16 optimization is set convert Fp32 network to Fp16
     if (options.m_ReduceFp32ToFp16)