diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-08-05 17:29:32 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-08-06 10:13:00 +0000 |
commit | 84e025a90dd4f98e03608ac6f9a58866eeb04687 (patch) | |
tree | 82e186b3db604db68e9493c08b247e9cddd5afd8 /src/armnn | |
parent | 81beae3a870004795275e9266bc43d845b9f78db (diff) | |
download | armnn-84e025a90dd4f98e03608ac6f9a58866eeb04687.tar.gz |
Infer tensor shapes before graph optimization
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I2a7a9b7b21848e2b3258788f06b01159f95a0d6c
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Network.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index a29ce83c5a..22a71c4923 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1594,13 +1594,13 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, // 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(), |