diff options
Diffstat (limited to 'src/backends/cl/workloads/ClSoftmaxWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClSoftmaxWorkload.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClSoftmaxWorkload.hpp b/src/backends/cl/workloads/ClSoftmaxWorkload.hpp index 158bf46c32..1742c60511 100644 --- a/src/backends/cl/workloads/ClSoftmaxWorkload.hpp +++ b/src/backends/cl/workloads/ClSoftmaxWorkload.hpp @@ -23,8 +23,10 @@ arm_compute::Status ClSoftmaxWorkloadValidate(const TensorInfo& input, class ClSoftmaxWorkload : public BaseWorkload<SoftmaxQueueDescriptor> { public: - ClSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor, const WorkloadInfo& info, - std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); + ClSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor, + const WorkloadInfo& info, + std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager, + const arm_compute::CLCompileContext& clCompileContext); void Execute() const override; private: |