ArmNN
NotReleased
NeonSoftmaxFloatWorkload.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
NeonSoftmaxFloatWorkload
:
public
FloatWorkload
<SoftmaxQueueDescriptor>
19
{
20
public
:
21
NeonSoftmaxFloatWorkload
(
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::BoostLogSeverityMapping::info
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::NeonSoftmaxFloatWorkload
Definition:
NeonSoftmaxFloatWorkload.hpp:18
armnn
Definition:
BackendHelper.hpp:11
armnn::NeonSoftmaxFloatWorkload::Execute
virtual void Execute() const override
Definition:
NeonSoftmaxFloatWorkload.cpp:32
armnn::SoftmaxQueueDescriptor
Definition:
WorkloadData.hpp:77
armnn::NeonSoftmaxFloatWorkload::NeonSoftmaxFloatWorkload
NeonSoftmaxFloatWorkload(const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
Definition:
NeonSoftmaxFloatWorkload.cpp:16
src
backends
neon
workloads
NeonSoftmaxFloatWorkload.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13