20.02
RefPadWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
backendsCommon/Workload.hpp
>
9
#include <
backendsCommon/WorkloadData.hpp
>
10
11
#include <
armnn/TypesUtils.hpp
>
12
13
namespace
armnn
14
{
15
16
template
<armnn::DataType DataType>
17
class
RefPadWorkload
:
public
TypedWorkload
<PadQueueDescriptor, DataType>
18
{
19
public
:
20
21
static
const
std::string&
GetName
()
22
{
23
static
const
std::string name = std::string(
"RefPad"
) +
GetDataTypeName
(
DataType
) +
"Workload"
;
24
return
name;
25
}
26
27
using
TypedWorkload<PadQueueDescriptor, DataType>::m_Data
;
28
using
TypedWorkload<PadQueueDescriptor, DataType>::TypedWorkload
;
29
30
void
Execute
()
const override
;
31
};
32
33
using
RefPadBFloat16Workload
=
RefPadWorkload<DataType::BFloat16>
;
34
using
RefPadFloat32Workload
=
RefPadWorkload<DataType::Float32>
;
35
using
RefPadFloat16Workload
=
RefPadWorkload<DataType::Float16>
;
36
using
RefPadQAsymm8Workload
=
RefPadWorkload<DataType::QAsymmU8>
;
37
using
RefPadQSymm16Workload
=
RefPadWorkload<DataType::QSymmS16>
;
38
39
}
//namespace armnn
WorkloadData.hpp
armnn::RefPadWorkload
Definition:
RefPadWorkload.hpp:17
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
armnn::RefPadWorkload::Execute
void Execute() const override
Definition:
RefPadWorkload.cpp:21
armnn::GetDataTypeName
constexpr const char * GetDataTypeName(DataType dataType)
Definition:
TypesUtils.hpp:168
armnn::RefPadWorkload::GetName
static const std::string & GetName()
Definition:
RefPadWorkload.hpp:21
TypesUtils.hpp
armnn::DataType
DataType
Definition:
Types.hpp:32
armnn::TypedWorkload
Definition:
Workload.hpp:52
Workload.hpp
src
backends
reference
workloads
RefPadWorkload.hpp
Generated on Fri Mar 13 2020 16:09:13 for ArmNN by
1.8.13