23.02
NeonSpaceToBatchNdWorkload.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/NESpaceToBatchLayer.h>
14
15
namespace
armnn
16
{
17
18
arm_compute::Status
NeonSpaceToBatchNdWorkloadValidate
(
const
TensorInfo& input,
19
const
TensorInfo& output,
20
const
SpaceToBatchNdDescriptor& descriptor);
21
22
class
NeonSpaceToBatchNdWorkload
:
public
NeonBaseWorkload
<SpaceToBatchNdQueueDescriptor>
23
{
24
public
:
25
using
NeonBaseWorkload<SpaceToBatchNdQueueDescriptor>::NeonBaseWorkload
;
26
27
NeonSpaceToBatchNdWorkload
(
const
SpaceToBatchNdQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
28
29
virtual
void
Execute
()
const override
;
30
31
private
:
32
mutable
std::unique_ptr<arm_compute::NESpaceToBatchLayer> m_Layer;
33
};
34
35
}
//namespace armnn
armnn::NeonSpaceToBatchNdWorkloadValidate
arm_compute::Status NeonSpaceToBatchNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor)
Definition:
NeonSpaceToBatchNdWorkload.cpp:20
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::NeonSpaceToBatchNdWorkload::Execute
virtual void Execute() const override
Definition:
NeonSpaceToBatchNdWorkload.cpp:84
Tensor.hpp
armnn::Status
Status
Definition:
Types.hpp:42
armnn::SpaceToBatchNdQueueDescriptor
Definition:
WorkloadData.hpp:380
armnn::NeonSpaceToBatchNdWorkload
Definition:
NeonSpaceToBatchNdWorkload.hpp:22
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
NeonBaseWorkload.hpp
Descriptors.hpp
armnn::NeonSpaceToBatchNdWorkload::NeonSpaceToBatchNdWorkload
NeonSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonSpaceToBatchNdWorkload.cpp:44
armnn::NeonBaseWorkload
Definition:
NeonBaseWorkload.hpp:13
armnn::BoostLogSeverityMapping::info
@ info
src
backends
neon
workloads
NeonSpaceToBatchNdWorkload.hpp
Generated on Wed Mar 22 2023 15:53:02 for ArmNN by
1.8.17