2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
8 #include <backendsCommon/Workload.hpp>
10 #include <arm_compute/runtime/CL/CLFunctions.h>
11 #include <arm_compute/runtime/MemoryManagerOnDemand.h>
18 class ClSoftmaxFloatWorkload : public FloatWorkload<SoftmaxQueueDescriptor>
21 ClSoftmaxFloatWorkload(const SoftmaxQueueDescriptor& descriptor, const WorkloadInfo& info,
22 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
23 void Execute() const override;
26 mutable arm_compute::CLSoftmaxLayer m_SoftmaxLayer;