10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 12 #include <arm_compute/runtime/CL/functions/CLSoftmaxLayer.h> 21 const TensorInfo& output,
28 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
29 const arm_compute::CLCompileContext& clCompileContext);
33 mutable arm_compute::CLLogSoftmaxLayer m_LogSoftmaxLayer;
arm_compute::Status ClLogSoftmaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor)
Copyright (c) 2021 ARM Limited and Contributors.
SoftmaxDescriptor LogSoftmaxDescriptor
A LogSoftmaxDescriptor for the LogSoftmaxLayer.
Contains information about TensorInfos of a layer.
ClLogSoftmaxWorkload(const LogSoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override