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 /CMakeLists.txt | |
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 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 9396316446..f55f391622 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -441,12 +441,12 @@ list(APPEND armnn_sources src/armnn/optimizations/ConvertFp32NetworkToFp16.hpp src/armnn/optimizations/FoldPadIntoConvolution2d.hpp src/armnn/optimizations/MovePermuteUp.hpp + src/armnn/optimizations/MoveTransposeUp.hpp src/armnn/optimizations/Optimization.hpp src/armnn/optimizations/OptimizeConsecutiveReshapes.hpp src/armnn/optimizations/OptimizeInverseConversions.hpp src/armnn/optimizations/OptimizeInversePermutes.hpp src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.hpp - src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.cpp src/armnn/optimizations/PermuteAsReshape.hpp src/armnn/optimizations/SquashEqualSiblings.hpp src/profiling/BufferManager.cpp @@ -619,12 +619,14 @@ if(BUILD_UNIT_TESTS) src/armnn/test/optimizations/Fp32NetworkToFp16ConverterTests.cpp src/armnn/test/optimizations/InsertDebugLayerTests.cpp src/armnn/test/optimizations/MovePermuteUpTests.cpp + src/armnn/test/optimizations/MoveTransposeUpTests.cpp src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp src/armnn/test/optimizations/OptimizeInverseConversionsTests.cpp src/armnn/test/optimizations/OptimizeInversePermutesTests.cpp src/armnn/test/optimizations/PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp src/armnn/test/optimizations/PermuteAsReshapeTests.cpp src/armnn/test/optimizations/SquashEqualSiblingsTests.cpp + src/armnn/test/optimizations/TransposeAsReshapeTests.cpp src/armnn/test/OptionalTest.cpp src/armnn/test/ProfilerTests.cpp src/armnn/test/ProfilingEventTest.cpp |