diff options
author | Mike Kelly <mike.kelly@arm.com> | 2020-11-05 15:44:56 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-11-08 22:50:50 +0000 |
commit | 90231b8c9f680d323e4b93dcd0820a47925e6d24 (patch) | |
tree | c34ace59ad3dd4757650a7b5d8f21a996c31cc88 /src/armnn/Network.cpp | |
parent | 82490eee66b733ef29b3697b21e10a4c8be95233 (diff) | |
download | armnn-90231b8c9f680d323e4b93dcd0820a47925e6d24.tar.gz |
IVGCVSW-5315 Create FuseBatchNorm class
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: Id0625c58dbeea79874bf986b70d136ed9390bf83
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 6578b8445f..347e39b4c8 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1055,7 +1055,10 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, FoldPadIntoConvolution2d(), PermuteAndBatchToSpaceAsDepthToSpace(), TransposeAndBatchToSpaceAsDepthToSpace(), - FuseBatchNormIntoConvolution2D())); + FuseBatchNormIntoConvolution2DFloat32(), + FuseBatchNormIntoConvolution2DFloat16(), + FuseBatchNormIntoDepthwiseConvolution2DFloat32(), + FuseBatchNormIntoDepthwiseConvolution2DFloat16())); // If Fp32 to Fp16 optimization is set convert Fp32 network to Fp16 if (options.m_ReduceFp32ToFp16) |