From fd313fef775ed210f8dab84452ea382a0b4164b0 Mon Sep 17 00:00:00 2001
From: David Monahan <David.Monahan@arm.com>
Date: Fri, 28 Jan 2022 13:06:35 +0000
Subject: IVGCVSW-6679 Add input and output workload slot pairs to
 LoadedNetwork

 * Added vectors to store the indexes of workload queues which contain inputs or outputs and their corresponding input / output slots

Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: Ide626726718216c7a778cce583da75af3ca2dc9a
---
 src/armnn/LoadedNetwork.hpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/armnn/LoadedNetwork.hpp')

diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp
index 53c35d61f6..9de6307938 100644
--- a/src/armnn/LoadedNetwork.hpp
+++ b/src/armnn/LoadedNetwork.hpp
@@ -197,6 +197,11 @@ private:
     std::unique_ptr<MemoryManager> m_ExternalMemoryManager;
 
     std::unordered_map<BackendId, bool> m_SupportsExternallyManagedMemory;
+
+    // A set of vectors to record the workload queue indexes and their corresponding Input/Output Slot indexes
+    // which are connected to Inputs and Outputs for the network.
+    std::vector<std::pair<unsigned int, unsigned int>> m_InputWorkloadSlotPairs;
+    std::vector<std::pair<unsigned int, unsigned int>> m_OutputWorkloadSlotPairs;
 };
 
 }
-- 
cgit v1.2.1