diff options
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(); |