ArmNN
NotReleased
NeonPadWorkload.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/WorkloadData.hpp
>
9
#include <
backendsCommon/Workload.hpp
>
10
#include <arm_compute/runtime/IFunction.h>
11
#include <arm_compute/core/Error.h>
12
13
namespace
armnn
{
14
15
class
NeonPadWorkload
:
public
BaseWorkload
<PadQueueDescriptor>
16
{
17
public
:
18
NeonPadWorkload
(
const
PadQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
19
20
void
Execute
()
const override
;
21
22
private
:
23
std::unique_ptr<arm_compute::IFunction> m_Layer;
24
};
25
26
arm_compute::Status
NeonPadWorkloadValidate
(
const
TensorInfo
& input,
27
const
TensorInfo
& output,
28
const
PadDescriptor
& descriptor);
29
30
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::NeonPadWorkload::Execute
void Execute() const override
Definition:
NeonPadWorkload.cpp:42
armnn::BoostLogSeverityMapping::info
armnn::TensorInfo
Definition:
Tensor.hpp:53
armnn::PadDescriptor
A PadDescriptor for the PadLayer.
Definition:
Descriptors.hpp:901
armnn::NeonPadWorkloadValidate
arm_compute::Status NeonPadWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const PadDescriptor &descriptor)
Definition:
NeonPadWorkload.cpp:48
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::NeonPadWorkload
Definition:
NeonPadWorkload.hpp:15
armnn::NeonPadWorkload::NeonPadWorkload
NeonPadWorkload(const PadQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonPadWorkload.cpp:19
armnn::PadQueueDescriptor
Definition:
WorkloadData.hpp:252
WorkloadData.hpp
armnn
Definition:
BackendHelper.hpp:11
armnn::BaseWorkload
Definition:
Workload.hpp:28
src
backends
neon
workloads
NeonPadWorkload.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13