aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LoadedNetwork.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r--src/armnn/LoadedNetwork.hpp5
1 files changed, 5 insertions, 0 deletions
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;
};
}