diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2020-03-03 12:39:09 +0000 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2020-03-03 15:05:30 +0000 |
| commit | 490b7becb8029ead26423b0d62e631a929e55d6c (patch) | |
| tree | 31148ace54164f62927062b662b2526f22a02e95 /src/armnn/optimizations/All.hpp | |
| parent | 4a9e24bfc51eec7e593470091fb7e6e435ae3991 (diff) | |
| download | armnn-490b7becb8029ead26423b0d62e631a929e55d6c.tar.gz | |
IVGCVSW-4375 Add support for Transpose to optimizations
* Changed some existing Permutation specific optimizations to also support Transpose
* Added MoveTransposeUp optimization
* Added TransposeAsReshape optimization
* Added tests for Transpose optimizations
* Added missing layer tests for Transpose
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I20d099b284861402ae94aaa5dbf34907327a485f
Diffstat (limited to 'src/armnn/optimizations/All.hpp')
| -rw-r--r-- | src/armnn/optimizations/All.hpp | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/src/armnn/optimizations/All.hpp b/src/armnn/optimizations/All.hpp index 4ea3f7f2d4..273c337665 100644 --- a/src/armnn/optimizations/All.hpp +++ b/src/armnn/optimizations/All.hpp @@ -4,14 +4,16 @@ // #pragma once +#include "AddDebug.hpp" #include "ConvertConstants.hpp" +#include "ConvertFp32NetworkToFp16.hpp" +#include "FoldPadIntoConvolution2d.hpp" +#include "MovePermuteUp.hpp" +#include "MoveTransposeUp.hpp" +#include "OptimizeConsecutiveReshapes.hpp" +#include "OptimizeInverseConversions.hpp" #include "OptimizeInversePermutes.hpp" #include "PermuteAsReshape.hpp" -#include "OptimizeConsecutiveReshapes.hpp" +#include "PermuteAndBatchToSpaceAsDepthToSpace.hpp" #include "SquashEqualSiblings.hpp" -#include "MovePermuteUp.hpp" -#include "OptimizeInverseConversions.hpp" -#include "ConvertFp32NetworkToFp16.hpp" -#include "AddDebug.hpp" -#include "FoldPadIntoConvolution2d.hpp" -#include "PermuteAndBatchToSpaceAsDepthToSpace.hpp"
\ No newline at end of file +#include "TransposeAsReshape.hpp"
\ No newline at end of file |