23.02
ClSpaceToBatchNdWorkload.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/Tensor.hpp
>
9
#include <
armnn/Descriptors.hpp
>
10
11
#include "
ClBaseWorkload.hpp
"
12
13
#include <arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h>
14
15
namespace
armnn
16
{
17
18
arm_compute::Status
ClSpaceToBatchNdWorkloadValidate
(
const
TensorInfo& input,
19
const
TensorInfo& output,
20
const
SpaceToBatchNdDescriptor& descriptor);
21
22
class
ClSpaceToBatchNdWorkload
:
public
ClBaseWorkload
<SpaceToBatchNdQueueDescriptor>
23
{
24
public
:
25
ClSpaceToBatchNdWorkload
(
const
SpaceToBatchNdQueueDescriptor
& descriptor,
26
const
WorkloadInfo
&
info
,
27
const
arm_compute::CLCompileContext& clCompileContext);
28
void
Execute
()
const override
;
29
30
private
:
31
mutable
arm_compute::CLSpaceToBatchLayer m_SpaceToBatchLayer;
32
};
33
34
}
//namespace armnn
35
ClBaseWorkload.hpp
armnn::ClSpaceToBatchNdWorkloadValidate
arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor)
Definition:
ClSpaceToBatchNdWorkload.cpp:23
armnn::ClSpaceToBatchNdWorkload::Execute
void Execute() const override
Definition:
ClSpaceToBatchNdWorkload.cpp:91
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
Tensor.hpp
armnn::Status
Status
Definition:
Types.hpp:42
armnn::ClBaseWorkload
Definition:
ClBaseWorkload.hpp:13
armnn::ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload
ClSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition:
ClSpaceToBatchNdWorkload.cpp:47
armnn::SpaceToBatchNdQueueDescriptor
Definition:
WorkloadData.hpp:380
armnn::ClSpaceToBatchNdWorkload
Definition:
ClSpaceToBatchNdWorkload.hpp:22
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
Descriptors.hpp
armnn::BoostLogSeverityMapping::info
@ info
src
backends
cl
workloads
ClSpaceToBatchNdWorkload.hpp
Generated on Wed Mar 22 2023 15:53:01 for ArmNN by
1.8.17