diff options
Diffstat (limited to 'include/armnn/IWorkingMemHandle.hpp')
-rw-r--r-- | include/armnn/IWorkingMemHandle.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/include/armnn/IWorkingMemHandle.hpp b/include/armnn/IWorkingMemHandle.hpp index 921b7e1f40..171fa3d81c 100644 --- a/include/armnn/IWorkingMemHandle.hpp +++ b/include/armnn/IWorkingMemHandle.hpp @@ -10,6 +10,8 @@ namespace armnn { +using NetworkId = int; + namespace experimental { @@ -20,6 +22,9 @@ class IWorkingMemHandle public: virtual ~IWorkingMemHandle() {}; + /// Returns the NetworkId of the Network that this IWorkingMemHandle works with. + virtual NetworkId GetNetworkId() = 0; + /// Allocate the backing memory required for execution. If this is not called, then allocation will be /// deferred to execution time. The mutex must be locked. virtual void Allocate() = 0; |