diff options
Diffstat (limited to 'include/armnn/backends/IWorkload.hpp')
-rw-r--r-- | include/armnn/backends/IWorkload.hpp | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/include/armnn/backends/IWorkload.hpp b/include/armnn/backends/IWorkload.hpp index 0bd8d2db75..a4827ebcdf 100644 --- a/include/armnn/backends/IWorkload.hpp +++ b/include/armnn/backends/IWorkload.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd. All rights reserved. +// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -9,6 +9,15 @@ namespace armnn { +namespace experimental +{ + +struct WorkingMemDescriptor; + +} // end experimental namespace + +using namespace armnn::experimental; + /// Workload interface to enqueue a layer computation. class IWorkload { public: @@ -18,9 +27,11 @@ public: virtual void Execute() const = 0; + virtual void ExecuteAsync(WorkingMemDescriptor& desc) = 0; + virtual profiling::ProfilingGuid GetGuid() const = 0; - virtual void RegisterDebugCallback(const DebugCallbackFunction & /*func*/) {} + virtual void RegisterDebugCallback(const DebugCallbackFunction& /*func*/) {} }; } //namespace armnn |