diff options
Diffstat (limited to 'include/armnn/IWorkingMemHandle.hpp')
| -rw-r--r-- | include/armnn/IWorkingMemHandle.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/include/armnn/IWorkingMemHandle.hpp b/include/armnn/IWorkingMemHandle.hpp index 171fa3d81c..6fb2f9fe5f 100644 --- a/include/armnn/IWorkingMemHandle.hpp +++ b/include/armnn/IWorkingMemHandle.hpp @@ -25,6 +25,9 @@ public: /// Returns the NetworkId of the Network that this IWorkingMemHandle works with. virtual NetworkId GetNetworkId() = 0; + /// Returns the InferenceId of the Inference that this IWorkingMemHandle works with. + virtual profiling::ProfilingGuid GetInferenceId() = 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; |