diff options
Diffstat (limited to 'src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp b/src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp index 4eb86eac8a..988408c9b7 100644 --- a/src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp +++ b/src/backends/cl/workloads/ClLogSoftmaxWorkload.hpp @@ -12,7 +12,7 @@ #include <arm_compute/runtime/CL/functions/CLSoftmaxLayer.h> -#include <armnn/backends/Workload.hpp> +#include "ClBaseWorkload.hpp" namespace armnn { @@ -21,7 +21,7 @@ arm_compute::Status ClLogSoftmaxWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const LogSoftmaxDescriptor& descriptor); -class ClLogSoftmaxWorkload : public BaseWorkload<LogSoftmaxQueueDescriptor> +class ClLogSoftmaxWorkload : public ClBaseWorkload<LogSoftmaxQueueDescriptor> { public: ClLogSoftmaxWorkload(const LogSoftmaxQueueDescriptor& descriptor, const WorkloadInfo& info, |