23.02
NeonPermuteWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
NeonBaseWorkload.hpp
"
9
#include <
armnn/backends/WorkloadData.hpp
>
10
#include <
neon/workloads/NeonWorkloadUtils.hpp
>
11
12
#include <
armnn/TypesUtils.hpp
>
13
#include <arm_compute/runtime/NEON/functions/NEPermute.h>
14
15
#include <string>
16
17
namespace
armnn
18
{
19
arm_compute::Status
NeonPermuteWorkloadValidate
(
const
TensorInfo& input,
const
TensorInfo& output,
20
const
PermuteDescriptor& descriptor);
21
22
class
NeonPermuteWorkload
:
public
NeonBaseWorkload
<PermuteQueueDescriptor>
23
{
24
public
:
25
static
const
std::string&
GetName
()
26
{
27
static
const
std::string name = std::string(
"NeonPermuteWorkload"
);
28
return
name;
29
}
30
31
NeonPermuteWorkload
(
const
PermuteQueueDescriptor
& descriptor,
const
WorkloadInfo
& info);
32
void
Execute
()
const override
;
33
34
private
:
35
using
BaseWorkload<PermuteQueueDescriptor>::m_Data
;
36
mutable
arm_compute::NEPermute m_PermuteFunction;
37
};
38
39
}
// namespace armnn
armnn::PermuteQueueDescriptor
Definition:
WorkloadData.hpp:186
armnn::NeonPermuteWorkload::NeonPermuteWorkload
NeonPermuteWorkload(const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonPermuteWorkload.cpp:27
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::BaseWorkload
Definition:
Workload.hpp:33
NeonWorkloadUtils.hpp
armnn::Status
Status
Definition:
Types.hpp:42
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
armnn::NeonPermuteWorkloadValidate
arm_compute::Status NeonPermuteWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const PermuteDescriptor &descriptor)
Definition:
NeonPermuteWorkload.cpp:15
armnn::NeonPermuteWorkload::Execute
void Execute() const override
Definition:
NeonPermuteWorkload.cpp:49
NeonBaseWorkload.hpp
armnn::NeonPermuteWorkload
Definition:
NeonPermuteWorkload.hpp:22
armnn::NeonBaseWorkload
Definition:
NeonBaseWorkload.hpp:13
armnn::NeonPermuteWorkload::GetName
static const std::string & GetName()
Definition:
NeonPermuteWorkload.hpp:25
TypesUtils.hpp
WorkloadData.hpp
src
backends
neon
workloads
NeonPermuteWorkload.hpp
Generated on Wed Mar 22 2023 15:53:02 for ArmNN by
1.8.17