diff options
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r-- | src/armnn/LoadedNetwork.hpp | 5 |
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; }; } |