20.11
NeonReshapeWorkload.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
#include <
neon/workloads/NeonWorkloadUtils.hpp
>
11
12
#include <
armnn/TypesUtils.hpp
>
13
#include <arm_compute/runtime/IFunction.h>
14
15
#include <memory>
16
17
namespace
armnn
18
{
19
20
arm_compute::Status
NeonReshapeWorkloadValidate
(
const
TensorInfo& input,
const
TensorInfo& output);
21
22
class
NeonReshapeWorkload
:
public
BaseWorkload
<ReshapeQueueDescriptor>
23
{
24
public
:
25
NeonReshapeWorkload
(
const
ReshapeQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
26
27
virtual
void
Execute
()
const override
;
28
29
private
:
30
std::unique_ptr<arm_compute::IFunction> m_Layer;
31
};
32
33
}
//namespace armnn
armnn::NeonReshapeWorkload::NeonReshapeWorkload
NeonReshapeWorkload(const ReshapeQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonReshapeWorkload.cpp:26
WorkloadData.hpp
armnn::NeonReshapeWorkload
Definition:
NeonReshapeWorkload.hpp:22
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
TypesUtils.hpp
armnn::BaseWorkload
Definition:
Workload.hpp:28
armnn::NeonReshapeWorkloadValidate
arm_compute::Status NeonReshapeWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition:
NeonReshapeWorkload.cpp:17
armnn::Status
Status
enumeration
Definition:
Types.hpp:26
NeonWorkloadUtils.hpp
armnn::NeonReshapeWorkload::Execute
virtual void Execute() const override
Definition:
NeonReshapeWorkload.cpp:40
Workload.hpp
armnn::BoostLogSeverityMapping::info
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition:
WorkloadInfo.hpp:16
armnn::ReshapeQueueDescriptor
Definition:
WorkloadData.hpp:367
src
backends
neon
workloads
NeonReshapeWorkload.hpp
Generated on Wed Nov 25 2020 09:59:13 for ArmNN by
1.8.13