24.02
ClQuantizeWorkload.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
10
#include <arm_compute/runtime/CL/functions/CLQuantizationLayer.h>
11
12
namespace
armnn
13
{
14
15
arm_compute::Status
ClQuantizeWorkloadValidate
(
const
TensorInfo& input,
16
const
TensorInfo& output);
17
18
class
ClQuantizeWorkload
:
public
ClBaseWorkload
<QuantizeQueueDescriptor>
19
{
20
public
:
21
ClQuantizeWorkload
(
const
QuantizeQueueDescriptor
& descriptor,
22
const
WorkloadInfo
&
info
,
23
const
arm_compute::CLCompileContext& clCompileContext);
24
void
Execute
()
const override
;
25
26
private
:
27
mutable
arm_compute::CLQuantizationLayer m_Layer;
28
};
29
30
}
//namespace armnn
armnn::ClQuantizeWorkloadValidate
arm_compute::Status ClQuantizeWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition:
ClQuantizeWorkload.cpp:22
armnn::QuantizeQueueDescriptor
Definition:
WorkloadData.hpp:299
armnn::ClBaseWorkload
Definition:
ClBaseWorkload.hpp:13
armnn::ClQuantizeWorkload::ClQuantizeWorkload
ClQuantizeWorkload(const QuantizeQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition:
ClQuantizeWorkload.cpp:32
armnn::ClQuantizeWorkload
Definition:
ClQuantizeWorkload.hpp:18
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
armnn::ClQuantizeWorkload::Execute
void Execute() const override
Definition:
ClQuantizeWorkload.cpp:48
armnn::BoostLogSeverityMapping::info
@ info
armnn::Status
Status
Definition:
Types.hpp:42
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
ClBaseWorkload.hpp
src
backends
cl
workloads
ClQuantizeWorkload.hpp
Generated on Wed Feb 14 2024 16:36:16 for Arm NN by
1.8.17