diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-12 15:54:58 +0100 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-05-17 15:27:23 +0000 |
commit | 8bd536064f32b1ce521875e2bc8dd24044c4f5c3 (patch) | |
tree | 76bbaccf935d33b1ea57daf9cd32f58f8d04f903 /src/armnn/Network.cpp | |
parent | 9734f0911cdf68e1a7ba8b38167bb958436bc708 (diff) | |
download | armnn-8bd536064f32b1ce521875e2bc8dd24044c4f5c3.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; } |