ArmNN
NotReleased
ClSoftmaxUint8Workload.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/CLSoftmaxLayer.h>
11
#include <arm_compute/runtime/MemoryManagerOnDemand.h>
12
13
#include <memory>
14
15
namespace
armnn
16
{
17
// Softmax
18
class
ClSoftmaxUint8Workload
:
public
Uint8Workload
<SoftmaxQueueDescriptor>
19
{
20
public
:
21
ClSoftmaxUint8Workload
(
const
SoftmaxQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
,
22
std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
23
24
void
Execute
()
const override
;
25
private
:
26
27
mutable
arm_compute::CLSoftmaxLayer m_SoftmaxLayer;
28
};
29
30
}
//namespace armnn
31
armnn::TypedWorkload
Definition:
Workload.hpp:52
armnn::BoostLogSeverityMapping::info
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::ClSoftmaxUint8Workload::ClSoftmaxUint8Workload
ClSoftmaxUint8Workload(const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
Definition:
ClSoftmaxUint8Workload.cpp:16
armnn::ClSoftmaxUint8Workload
Definition:
ClSoftmaxUint8Workload.hpp:18
armnn
Definition:
BackendHelper.hpp:11
armnn::ClSoftmaxUint8Workload::Execute
void Execute() const override
Definition:
ClSoftmaxUint8Workload.cpp:40
armnn::SoftmaxQueueDescriptor
Definition:
WorkloadData.hpp:77
src
backends
cl
workloads
ClSoftmaxUint8Workload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13