From 33f8987b5a437e8fc35cb38dac18007a8d92db6e Mon Sep 17 00:00:00 2001
From: Matthew Sloyan <matthew.sloyan@arm.com>
Date: Wed, 30 Jun 2021 10:20:17 +0100
Subject: IVGCVSW-6161 ConstTensorsAsInput: Optimizer - Redirect ConstTensor
 layer members

 * Optimization that searches for layers with ConstantLayers as inputs.
 * The layer member variables are then redirected to these ConstantLayers.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I24a2bf0e8575b808343e0bbe3897b344e94796ad
---
 src/armnn/Network.cpp | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/armnn/Network.cpp')

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 74c195f676..d340f021e2 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1618,6 +1618,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork,
                                                 PermuteAsReshape(),
                                                 TransposeAsReshape(),
                                                 OptimizeConsecutiveReshapes(),
+                                                RedirectMembersToConstantInputs(),
                                                 FoldPadIntoConvolution2d(),
                                                 FoldPadIntoDepthwiseConvolution2d(),
                                                 FoldPadIntoPooling2d(),
-- 
cgit v1.2.1