diff options
| author | David Monahan <David.Monahan@arm.com> | 2022-01-28 13:06:35 +0000 |
|---|---|---|
| committer | David Monahan <David.Monahan@arm.com> | 2022-01-28 13:06:35 +0000 |
| commit | fd313fef775ed210f8dab84452ea382a0b4164b0 (patch) | |
| tree | 0b46fc1314b3aa404933268822fe8dbdfa6441f1 /src/armnn/LoadedNetwork.hpp | |
| parent | 2d24e5e615379f5cfce937a4e3bea06ac9ad2ec5 (diff) | |
| download | armnn-fd313fef775ed210f8dab84452ea382a0b4164b0.tar.gz | |
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
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; }; } |