diff options
Diffstat (limited to 'include/armnn/backends/IWorkload.hpp')
-rw-r--r-- | include/armnn/backends/IWorkload.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/include/armnn/backends/IWorkload.hpp b/include/armnn/backends/IWorkload.hpp index a4827ebcdf..d63e0acc72 100644 --- a/include/armnn/backends/IWorkload.hpp +++ b/include/armnn/backends/IWorkload.hpp @@ -31,6 +31,12 @@ public: virtual profiling::ProfilingGuid GetGuid() const = 0; + // Replace input tensor handle with the given TensorHandle + virtual void ReplaceInputTensorHandle(ITensorHandle* /*input*/, unsigned int /*slot*/) = 0; + + // Replace output tensor handle with the given TensorHandle + virtual void ReplaceOutputTensorHandle(ITensorHandle* /*output*/, unsigned int /*slot*/) = 0; + virtual void RegisterDebugCallback(const DebugCallbackFunction& /*func*/) {} }; |