23.05
NeonBatchToSpaceNdWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
armnn/Tensor.hpp
>
9
#include <
armnn/Descriptors.hpp
>
10
11
#include "
NeonBaseWorkload.hpp
"
12
13
#include <arm_compute/runtime/NEON/functions/NEBatchToSpaceLayer.h>
14
15
namespace
armnn
16
{
17
18
arm_compute::Status
NeonBatchToSpaceNdWorkloadValidate
(
const
TensorInfo& input,
19
const
TensorInfo& output,
20
const
BatchToSpaceNdDescriptor& descriptor);
21
22
class
NeonBatchToSpaceNdWorkload
:
public
NeonBaseWorkload
<BatchToSpaceNdQueueDescriptor>
23
{
24
public
:
25
using
NeonBaseWorkload<BatchToSpaceNdQueueDescriptor>::NeonBaseWorkload
;
26
27
NeonBatchToSpaceNdWorkload
(
const
BatchToSpaceNdQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
28
29
virtual
void
Execute
()
const override
;
30
31
private
:
32
mutable
std::unique_ptr<arm_compute::NEBatchToSpaceLayer> m_Layer;
33
};
34
35
}
armnn::NeonBatchToSpaceNdWorkload
Definition:
NeonBatchToSpaceNdWorkload.hpp:22
armnn::NeonBatchToSpaceNdWorkload::Execute
virtual void Execute() const override
Definition:
NeonBatchToSpaceNdWorkload.cpp:73
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::NeonBatchToSpaceNdWorkloadValidate
arm_compute::Status NeonBatchToSpaceNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &descriptor)
Definition:
NeonBatchToSpaceNdWorkload.cpp:20
Tensor.hpp
armnn::Status
Status
Definition:
Types.hpp:42
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
NeonBaseWorkload.hpp
armnn::NeonBatchToSpaceNdWorkload::NeonBatchToSpaceNdWorkload
NeonBatchToSpaceNdWorkload(const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonBatchToSpaceNdWorkload.cpp:41
armnn::BatchToSpaceNdQueueDescriptor
Definition:
WorkloadData.hpp:457
Descriptors.hpp
armnn::NeonBaseWorkload
Definition:
NeonBaseWorkload.hpp:13
armnn::BoostLogSeverityMapping::info
@ info
src
backends
neon
workloads
NeonBatchToSpaceNdWorkload.hpp
Generated on Thu May 18 2023 10:35:42 for ArmNN by
1.8.17