diff options
Diffstat (limited to 'src/armnn/WorkingMemHandle.hpp')
-rw-r--r-- | src/armnn/WorkingMemHandle.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/WorkingMemHandle.hpp b/src/armnn/WorkingMemHandle.hpp index 92b0acaec3..5ccb2b2342 100644 --- a/src/armnn/WorkingMemHandle.hpp +++ b/src/armnn/WorkingMemHandle.hpp @@ -38,6 +38,11 @@ public: return m_NetworkId; } + profiling::ProfilingGuid GetInferenceId() override + { + return m_InferenceId; + } + /// 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. void Allocate() override; @@ -87,6 +92,7 @@ private: bool m_IsAllocated; std::mutex m_Mutex; + profiling::ProfilingGuid m_InferenceId; }; } // end experimental namespace |