commit 84e025a90dd4f98e03608ac6f9a58866eeb04687
author Finn Williams <Finn.Williams@arm.com> Thu Aug 05 17:29:32 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Fri Aug 06 10:13:00 2021 +0000
tree 82e186b3db604db68e9493c08b247e9cddd5afd8
parent 81beae3a870004795275e9266bc43d845b9f78db

Infer tensor shapes before graph optimization

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I2a7a9b7b21848e2b3258788f06b01159f95a0d6c

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index a29ce83..22a71c4 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1594,13 +1594,13 @@
     // Get the optimized graph
     Graph& optGraph = optNetObjPtr->pOptimizedNetworkImpl->GetGraph();
 
+    // Infer the tensor infos for all output slots. Throws an exception on failure
+    optGraph.InferTensorInfos();
+
     // 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
     Optimizer::Pass(optGraph, MakeOptimizations(SquashEqualPermuteSiblings(),
                                                 SquashEqualTransposeSiblings(),