21.08
IWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020 Arm Ltd and Contributors. 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
namespace
experimental
13
{
14
15
struct
WorkingMemDescriptor
;
16
17
}
// end experimental namespace
18
19
using namespace
armnn::experimental
;
20
21
/// Workload interface to enqueue a layer computation.
22
class
IWorkload
{
23
public
:
24
virtual
~IWorkload
() {}
25
26
virtual
void
PostAllocationConfigure() = 0;
27
28
virtual
void
Execute()
const
= 0;
29
30
virtual
void
ExecuteAsync(
WorkingMemDescriptor
& desc) = 0;
31
32
virtual
profiling::ProfilingGuid GetGuid()
const
= 0;
33
34
virtual
void
RegisterDebugCallback
(
const
DebugCallbackFunction
&
/*func*/
) {}
35
};
36
37
}
//namespace armnn
armnn::IWorkload::RegisterDebugCallback
virtual void RegisterDebugCallback(const DebugCallbackFunction &)
Definition:
IWorkload.hpp:34
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_software_tools.dox:6
armnn::DebugCallbackFunction
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:321
armnn::experimental::WorkingMemDescriptor
Definition:
WorkingMemDescriptor.hpp:18
Types.hpp
armnn::IWorkload::~IWorkload
virtual ~IWorkload()
Definition:
IWorkload.hpp:24
armnn::IWorkload
Workload interface to enqueue a layer computation.
Definition:
IWorkload.hpp:22
armnn::experimental
Definition:
IWorkload.hpp:12
include
armnn
backends
IWorkload.hpp
Generated on Tue Aug 24 2021 16:18:19 for ArmNN by
1.8.13