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