ArmNN
NotReleased
ClDequantizeWorkload.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/CLDequantizationLayer.h>
11
12
namespace
armnn
13
{
14
15
arm_compute::Status
ClDequantizeWorkloadValidate
(
const
TensorInfo& input,
const
TensorInfo& output);
16
17
class
ClDequantizeWorkload
:
public
BaseWorkload
<DequantizeQueueDescriptor>
18
{
19
public
:
20
ClDequantizeWorkload
(
const
DequantizeQueueDescriptor
& descriptor,
const
WorkloadInfo
& workloadInfo);
21
22
void
Execute
()
const override
;
23
24
private
:
25
mutable
std::unique_ptr<arm_compute::CLDequantizationLayer> m_Layer;
26
};
27
28
}
// namespace armnn
armnn::DequantizeQueueDescriptor
Definition:
WorkloadData.hpp:474
armnn::Status
Status
Definition:
Types.hpp:26
armnn::ClDequantizeWorkload
Definition:
ClDequantizeWorkload.hpp:17
armnn::ClDequantizeWorkloadValidate
arm_compute::Status ClDequantizeWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition:
ClDequantizeWorkload.cpp:23
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn
Definition:
BackendHelper.hpp:11
armnn::ClDequantizeWorkload::ClDequantizeWorkload
ClDequantizeWorkload(const DequantizeQueueDescriptor &descriptor, const WorkloadInfo &workloadInfo)
Definition:
ClDequantizeWorkload.cpp:31
armnn::ClDequantizeWorkload::Execute
void Execute() const override
Definition:
ClDequantizeWorkload.cpp:48
armnn::BaseWorkload
Definition:
Workload.hpp:28
src
backends
cl
workloads
ClDequantizeWorkload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13