10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 12 #include <arm_compute/runtime/CL/functions/CLSoftmaxLayer.h> 20 const TensorInfo& output,
21 const SoftmaxDescriptor& descriptor);
27 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
31 mutable arm_compute::CLSoftmaxLayer m_SoftmaxLayer;
void Execute() const override
Copyright (c) 2020 ARM Limited.
arm_compute::Status ClSoftmaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor)
ClSoftmaxWorkload(const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
Contains information about inputs and outputs to a layer.