diff options
| author | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-12 15:54:58 +0100 |
|---|---|---|
| committer | Ryan OShea <ryan.oshea3@arm.com> | 2022-05-19 11:05:30 +0100 |
| commit | f6ae98331f7d08e5ab764ab70a168a523fdd5404 (patch) | |
| tree | b085b6ecb066e20090d1ac16d31b30ad62690dbe /src/armnn/Network.cpp | |
| parent | 2cddc72f7aa1eab43c69250e608d662909383ba7 (diff) | |
| download | armnn-f6ae98331f7d08e5ab764ab70a168a523fdd5404.tar.gz | |
IVGCVSW-6126 ConstTensorsAsInput: Conv2d - Backends
!android-nn-driver:7477
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ibf633ccccc385bd980934ff829407d21981323ef
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
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; } |