diff options
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index a668274c4d..cf9a138084 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -818,7 +818,8 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, MovePermuteUp(), PermuteAsReshape(), OptimizeConsecutiveReshapes(), - FoldPadIntoConvolution2d())); + FoldPadIntoConvolution2d(), + PermuteAndBatchToSpaceAsDepthToSpace())); // Infer the tensor infos for all output slots. Throws an exception on failure optGraph.InferTensorInfos(); |