diff options
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 17813a8983..cd5f369271 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1038,11 +1038,14 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, // Get the optimized graph Graph& optGraph = optNetObjPtr->GetGraph(); + // Perform AddBroadcastReshapeLayer optimisation + using namespace optimizations; + Optimizer::Pass(optGraph, MakeOptimizations(AddBroadcastReshapeLayer())); + // Infer the tensor infos for all output slots. Throws an exception on failure optGraph.InferTensorInfos(); // Perform optimisation passes - using namespace optimizations; Optimizer::Pass(optGraph, MakeOptimizations(SquashEqualPermuteSiblings(), SquashEqualTransposeSiblings(), SquashEqualReshapeSiblings(), |