From 84e025a90dd4f98e03608ac6f9a58866eeb04687 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Thu, 5 Aug 2021 17:29:32 +0100 Subject: Infer tensor shapes before graph optimization Signed-off-by: Finn Williams Change-Id: I2a7a9b7b21848e2b3258788f06b01159f95a0d6c --- src/armnn/Network.cpp | 6 +++--- 1 file 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(), -- cgit v1.2.1