22.02
RefTransposeWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
RefBaseWorkload.hpp
"
9
10
#include <
armnn/TypesUtils.hpp
>
11
12
namespace
armnn
13
{
14
15
template
<armnn::DataType DataType>
16
class
RefTransposeWorkload
:
public
TypedWorkload
<TransposeQueueDescriptor, DataType>
17
{
18
public
:
19
static
const
std::string&
GetName
()
20
{
21
static
const
std::string name = std::string(
"RefTranspose"
) +
GetDataTypeName
(
DataType
) +
"Workload"
;
22
return
name;
23
}
24
25
using
TypedWorkload<TransposeQueueDescriptor, DataType>::m_Data
;
26
using
TypedWorkload<TransposeQueueDescriptor, DataType>::TypedWorkload
;
27
void
Execute
()
const override
;
28
void
ExecuteAsync
(
WorkingMemDescriptor
& workingMemDescriptor)
override
;
29
private
:
30
void
Execute
(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs)
const
;
31
};
32
33
using
RefTransposeBFloat16Workload
=
RefTransposeWorkload<DataType::BFloat16>
;
34
using
RefTransposeFloat16Workload
=
RefTransposeWorkload<DataType::Float16>
;
35
using
RefTransposeFloat32Workload
=
RefTransposeWorkload<DataType::Float32>
;
36
using
RefTransposeQAsymmS8Workload
=
RefTransposeWorkload<DataType::QAsymmS8>
;
37
using
RefTransposeQAsymm8Workload
=
RefTransposeWorkload<DataType::QAsymmU8>
;
38
using
RefTransposeQSymm16Workload
=
RefTransposeWorkload<DataType::QSymmS16>
;
39
40
}
//namespace armnn
armnn::RefTransposeWorkload
Definition:
RefTransposeWorkload.hpp:16
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::GetDataTypeName
constexpr const char * GetDataTypeName(DataType dataType)
Definition:
TypesUtils.hpp:202
armnn::experimental::WorkingMemDescriptor
Definition:
WorkingMemDescriptor.hpp:18
TypesUtils.hpp
armnn::DataType
DataType
Definition:
Types.hpp:35
armnn::TypedWorkload
Definition:
Workload.hpp:86
armnn::RefTransposeWorkload::Execute
void Execute() const override
Definition:
RefTransposeWorkload.cpp:17
armnn::RefTransposeWorkload::ExecuteAsync
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
Definition:
RefTransposeWorkload.cpp:23
RefBaseWorkload.hpp
armnn::RefTransposeWorkload::GetName
static const std::string & GetName()
Definition:
RefTransposeWorkload.hpp:19
src
backends
reference
workloads
RefTransposeWorkload.hpp
Generated on Wed Mar 9 2022 12:01:00 for ArmNN by
1.8.13