ArmNN
 20.02
IWorkload.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2020 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <armnn/Types.hpp>
8 
9 namespace armnn
10 {
11 
12 /// Workload interface to enqueue a layer computation.
13 class IWorkload {
14 public:
15  virtual ~IWorkload() {}
16 
17  virtual void PostAllocationConfigure() = 0;
18 
19  virtual void Execute() const = 0;
20 
21  virtual profiling::ProfilingGuid GetGuid() const = 0;
22 
23  virtual void RegisterDebugCallback(const DebugCallbackFunction & /*func*/) {}
24 };
25 
26 } //namespace armnn
virtual profiling::ProfilingGuid GetGuid() const =0
virtual void RegisterDebugCallback(const DebugCallbackFunction &)
Definition: IWorkload.hpp:23
virtual void PostAllocationConfigure()=0
Copyright (c) 2020 ARM Limited.
std::function< void(LayerGuid guid, unsigned int slotIndex, ITensorHandle *tensorHandle)> DebugCallbackFunction
Define the type of callback for the Debug layer to call.
Definition: Types.hpp:244
virtual ~IWorkload()
Definition: IWorkload.hpp:15
Workload interface to enqueue a layer computation.
Definition: IWorkload.hpp:13
virtual void Execute() const =0