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/Network.cpp | |
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/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 55bf51af00..50a7df6662 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -910,13 +910,18 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, // Perform optimisation passes using namespace optimizations; Optimizer::Pass(optGraph, MakeOptimizations(SquashEqualPermuteSiblings(), + SquashEqualTransposeSiblings(), SquashEqualReshapeSiblings(), OptimizeInversePermutes(), + OptimizeInverseTransposes(), MovePermuteUp(), + MoveTransposeUp(), PermuteAsReshape(), + TransposeAsReshape(), OptimizeConsecutiveReshapes(), FoldPadIntoConvolution2d(), - PermuteAndBatchToSpaceAsDepthToSpace())); + PermuteAndBatchToSpaceAsDepthToSpace(), + TransposeAndBatchToSpaceAsDepthToSpace())); // Infer the tensor infos for all output slots. Throws an exception on failure optGraph.InferTensorInfos(); |