ArmNN
NotReleased
ClReshapeWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
backendsCommon/Workload.hpp
>
9
10
#include <arm_compute/runtime/CL/functions/CLReshapeLayer.h>
11
12
namespace
armnn
13
{
14
15
arm_compute::Status
ClReshapeWorkloadValidate
(
const
TensorInfo& input,
16
const
TensorInfo& output);
17
18
class
ClReshapeWorkload
:
public
BaseWorkload
<ReshapeQueueDescriptor>
19
{
20
public
:
21
ClReshapeWorkload
(
const
ReshapeQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
22
23
void
Execute
()
const override
;
24
25
private
:
26
mutable
arm_compute::CLReshapeLayer m_Layer;
27
};
28
29
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::ReshapeQueueDescriptor
Definition:
WorkloadData.hpp:338
armnn::BoostLogSeverityMapping::info
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::ClReshapeWorkload::ClReshapeWorkload
ClReshapeWorkload(const ReshapeQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
ClReshapeWorkload.cpp:24
armnn
Definition:
BackendHelper.hpp:11
armnn::ClReshapeWorkload::Execute
void Execute() const override
Definition:
ClReshapeWorkload.cpp:35
armnn::ClReshapeWorkloadValidate
arm_compute::Status ClReshapeWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition:
ClReshapeWorkload.cpp:15
armnn::BaseWorkload
Definition:
Workload.hpp:28
armnn::ClReshapeWorkload
Definition:
ClReshapeWorkload.hpp:18
src
backends
cl
workloads
ClReshapeWorkload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13