24.08
NeonSpaceToDepthWorkload.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 <
armnn/TypesUtils.hpp
>
9
10
#include "
NeonBaseWorkload.hpp
"
11
12
#include <arm_compute/runtime/NEON/functions/NESpaceToDepthLayer.h>
13
14
namespace
armnn
15
{
16
17
arm_compute::Status
NeonSpaceToDepthWorkloadValidate
(
const
TensorInfo& input,
18
const
TensorInfo& output,
19
const
SpaceToDepthDescriptor& descriptor);
20
21
class
NeonSpaceToDepthWorkload
:
public
NeonBaseWorkload
<SpaceToDepthQueueDescriptor>
22
{
23
public
:
24
using
NeonBaseWorkload<SpaceToDepthQueueDescriptor>::NeonBaseWorkload
;
25
NeonSpaceToDepthWorkload
(
const
SpaceToDepthQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
26
virtual
void
Execute
()
const override
;
27
private
:
28
mutable
std::unique_ptr<arm_compute::NESpaceToDepthLayer> m_Layer;
29
};
30
31
}
//namespace armnn
armnn::SpaceToDepthQueueDescriptor
Definition:
WorkloadData.hpp:390
TypesUtils.hpp
NeonBaseWorkload.hpp
armnn::NeonSpaceToDepthWorkload::NeonSpaceToDepthWorkload
NeonSpaceToDepthWorkload(const SpaceToDepthQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonSpaceToDepthWorkload.cpp:32
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
armnn::NeonSpaceToDepthWorkloadValidate
arm_compute::Status NeonSpaceToDepthWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToDepthDescriptor &descriptor)
Definition:
NeonSpaceToDepthWorkload.cpp:19
armnn::BoostLogSeverityMapping::info
@ info
armnn::NeonSpaceToDepthWorkload
Definition:
NeonSpaceToDepthWorkload.hpp:21
armnn::NeonSpaceToDepthWorkload::Execute
virtual void Execute() const override
Definition:
NeonSpaceToDepthWorkload.cpp:59
armnn::Status
Status
Definition:
Types.hpp:42
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::NeonBaseWorkload
Definition:
NeonBaseWorkload.hpp:13
src
backends
neon
workloads
NeonSpaceToDepthWorkload.hpp
Generated on Wed Aug 28 2024 14:31:51 for Arm NN by
1.8.17