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 /Android.mk | |
| 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 'Android.mk')
| -rw-r--r-- | Android.mk | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Android.mk b/Android.mk index e423f25d7c..e29f0f913e 100644 --- a/Android.mk +++ b/Android.mk @@ -95,7 +95,6 @@ LOCAL_SRC_FILES := \ src/armnn/NetworkUtils.cpp \ src/armnn/Observable.cpp \ src/armnn/Optimizer.cpp \ - src/armnn/optimizations/PermuteAndBatchToSpaceAsDepthToSpace.cpp \ src/armnn/OutputHandler.cpp \ src/armnn/ProfilingEvent.cpp \ src/armnn/Profiling.cpp \ @@ -352,6 +351,7 @@ LOCAL_SRC_FILES := \ 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/OptimizerTests.cpp \ src/armnn/test/OptionalTest.cpp \ src/armnn/test/ProfilerTests.cpp \ |