11 #include <arm_compute/core/Error.h> 12 #include <arm_compute/runtime/IFunction.h> 13 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 21 const TensorInfo& output,
28 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
29 virtual void Execute()
const override;
32 std::unique_ptr<arm_compute::IFunction> m_LogSoftmaxLayer;
NeonLogSoftmaxWorkload(const LogSoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
arm_compute::Status NeonLogSoftmaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const LogSoftmaxDescriptor &descriptor)
Copyright (c) 2020 ARM Limited.
virtual void Execute() const override
SoftmaxDescriptor LogSoftmaxDescriptor
A LogSoftmaxDescriptor for the LogSoftmaxLayer.
Contains information about inputs and outputs to a layer.