diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-22 14:01:46 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-22 14:38:31 +0000 |
commit | 6e9482013f41725ccca0767c0c5db9b29f77d981 (patch) | |
tree | 7b33e2cb17009c3103b5731f8d8ffa2b37484347 /include | |
parent | 2a304ede65afb3426dccbd742ac3f5b42e8fb04a (diff) | |
download | armnn-6e9482013f41725ccca0767c0c5db9b29f77d981.tar.gz |
IVGCVSW-2865 Extend IRuntime to add a new method RegisterDebugCallback(...)
* Made changes to LoadedNetwork and IWorkload to pass on the registered
callback function
Change-Id: I6ea10f2a299d6de8bf681c8ff36d3fbed1d6d887
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/IRuntime.hpp | 5 | ||||
-rw-r--r-- | include/armnn/Types.hpp | 5 |
2 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/IRuntime.hpp b/include/armnn/IRuntime.hpp index b977afe5e7..44864ce86d 100644 --- a/include/armnn/IRuntime.hpp +++ b/include/armnn/IRuntime.hpp @@ -83,6 +83,11 @@ public: /// @return A pointer to the requested profiler, or nullptr if not found. virtual const std::shared_ptr<IProfiler> GetProfiler(NetworkId networkId) const = 0; + /// Registers a callback function to debug layers performing custom computations on intermediate tensors. + /// @param networkId The id of the network to register the callback. + /// @param func callback function to pass to the debug layer. + virtual void RegisterDebugCallback(NetworkId networkId, const DebugCallbackFunction& func) = 0; + protected: ~IRuntime() {} }; diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 36e3c5b52a..693a050586 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -180,4 +180,9 @@ private: /// Define LayerGuid type. using LayerGuid = unsigned int; +class ITensorHandle; + +/// Define the type of callback for the debug layer to call +using DebugCallbackFunction = std::function<void(LayerGuid, unsigned int, ITensorHandle*)>; + } // namespace armnn |