commit 8bd536064f32b1ce521875e2bc8dd24044c4f5c3
author Cathal Corbett <cathal.corbett@arm.com> Thu May 12 15:54:58 2022 +0100
committer Cathal Corbett <cathal.corbett@arm.com> Tue May 17 15:27:23 2022 +0000
tree 76bbaccf935d33b1ea57daf9cd32f58f8d04f903
parent 9734f0911cdf68e1a7ba8b38167bb958436bc708

IVGCVSW-6126 ConstTensorsAsInput: Conv2d - Backends

!android-nn-driver:7477

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ibf633ccccc385bd980934ff829407d21981323ef

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 479e57f..77ad5c4 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1832,6 +1832,9 @@
         ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "Optimizer_ConvertConstants");
         Optimizer::Pass(optGraph, MakeOptimizations(ConvertConstantsFloatToHalf()));
         Optimizer::Pass(optGraph, MakeOptimizations(ConvertConstantsHalfToFloat()));
+
+        // Once the constants are converted we can now safely call RedirectMembersToConstantInputs
+        Optimizer::Pass(optGraph, MakeOptimizations(RedirectMembersToConstantInputs()));
     }
     return optNet;
 }