23.05
ClBatchToSpaceNdWorkload.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 "
ClBaseWorkload.hpp
"
9
#include <arm_compute/runtime/CL/functions/CLBatchToSpaceLayer.h>
10
11
namespace
armnn
12
{
13
14
arm_compute::Status
ClBatchToSpaceNdWorkloadValidate
(
const
TensorInfo& input,
15
const
TensorInfo& output,
16
const
BatchToSpaceNdDescriptor& descriptor);
17
18
class
ClBatchToSpaceNdWorkload
:
public
ClBaseWorkload
<BatchToSpaceNdQueueDescriptor>
19
{
20
public
:
21
ClBatchToSpaceNdWorkload
(
const
BatchToSpaceNdQueueDescriptor
& descriptor,
22
const
WorkloadInfo
&
info
,
23
const
arm_compute::CLCompileContext& clCompileContext);
24
25
void
Execute
()
const override
;
26
27
private
:
28
29
mutable
arm_compute::CLBatchToSpaceLayer m_Layer;
30
};
31
32
}
//namespace armnn
armnn::ClBatchToSpaceNdWorkload::Execute
void Execute() const override
Definition:
ClBatchToSpaceNdWorkload.cpp:76
ClBaseWorkload.hpp
armnn::ClBatchToSpaceNdWorkloadValidate
arm_compute::Status ClBatchToSpaceNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &descriptor)
Definition:
ClBatchToSpaceNdWorkload.cpp:20
armnn::ClBatchToSpaceNdWorkload
Definition:
ClBatchToSpaceNdWorkload.hpp:18
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::Status
Status
Definition:
Types.hpp:42
armnn::ClBaseWorkload
Definition:
ClBaseWorkload.hpp:13
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
armnn::BatchToSpaceNdQueueDescriptor
Definition:
WorkloadData.hpp:457
armnn::ClBatchToSpaceNdWorkload::ClBatchToSpaceNdWorkload
ClBatchToSpaceNdWorkload(const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition:
ClBatchToSpaceNdWorkload.cpp:43
armnn::BoostLogSeverityMapping::info
@ info
src
backends
cl
workloads
ClBatchToSpaceNdWorkload.hpp
Generated on Thu May 18 2023 10:35:42 for ArmNN by
1.8.17