ArmNN
NotReleased
RefPermuteWorkload.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
10
#include <
armnn/TypesUtils.hpp
>
11
12
namespace
armnn
13
{
14
15
template
<armnn::DataType DataType>
16
class
RefPermuteWorkload
:
public
TypedWorkload
<PermuteQueueDescriptor, DataType>
17
{
18
public
:
19
static
const
std::string&
GetName
()
20
{
21
static
const
std::string name = std::string(
"RefPermute"
) +
GetDataTypeName
(
DataType
) +
"Workload"
;
22
return
name;
23
}
24
25
using
TypedWorkload<PermuteQueueDescriptor, DataType>::m_Data
;
26
using
TypedWorkload<PermuteQueueDescriptor, DataType>::TypedWorkload
;
27
void
Execute
()
const override
;
28
};
29
30
using
RefPermuteFloat16Workload
=
RefPermuteWorkload<DataType::Float16>
;
31
using
RefPermuteFloat32Workload
=
RefPermuteWorkload<DataType::Float32>
;
32
using
RefPermuteQAsymm8Workload
=
RefPermuteWorkload<DataType::QAsymmU8>
;
33
using
RefPermuteQSymm16Workload
=
RefPermuteWorkload<DataType::QSymmS16>
;
34
35
}
//namespace armnn
armnn::TypedWorkload
Definition:
Workload.hpp:52
armnn::RefPermuteWorkload::Execute
void Execute() const override
Definition:
RefPermuteWorkload.cpp:17
Workload.hpp
TypesUtils.hpp
armnn::RefPermuteWorkload
Definition:
RefPermuteWorkload.hpp:16
armnn::RefPermuteWorkload::GetName
static const std::string & GetName()
Definition:
RefPermuteWorkload.hpp:19
armnn
Definition:
BackendHelper.hpp:11
armnn::DataType
DataType
Definition:
Types.hpp:32
armnn::GetDataTypeName
constexpr const char * GetDataTypeName(DataType dataType)
Definition:
TypesUtils.hpp:165
src
backends
reference
workloads
RefPermuteWorkload.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13