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