From 8bd536064f32b1ce521875e2bc8dd24044c4f5c3 Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Thu, 12 May 2022 15:54:58 +0100 Subject: IVGCVSW-6126 ConstTensorsAsInput: Conv2d - Backends !android-nn-driver:7477 Signed-off-by: Cathal Corbett Change-Id: Ibf633ccccc385bd980934ff829407d21981323ef --- src/armnn/Network.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/armnn/Network.cpp') diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 479e57fc56..77ad5c4dc2 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1832,6 +1832,9 @@ IOptimizedNetworkPtr Optimize(const Graph& inGraph, 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; } -- cgit v1.2.1