diff options
Diffstat (limited to 'src/backends/cl/workloads/ClCastWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClCastWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClCastWorkload.hpp b/src/backends/cl/workloads/ClCastWorkload.hpp index d36c1b19e8..53a25bdc93 100644 --- a/src/backends/cl/workloads/ClCastWorkload.hpp +++ b/src/backends/cl/workloads/ClCastWorkload.hpp @@ -5,7 +5,7 @@ #pragma once -#include <armnn/backends/Workload.hpp> +#include "ClBaseWorkload.hpp" #include <arm_compute/core/Error.h> #include <arm_compute/runtime/CL/functions/CLCast.h> @@ -15,7 +15,7 @@ namespace armnn arm_compute::Status ClCastValidate(const TensorInfo& input, const TensorInfo& output); -class ClCastWorkload : public BaseWorkload<CastQueueDescriptor> +class ClCastWorkload : public ClBaseWorkload<CastQueueDescriptor> { public: ClCastWorkload(const CastQueueDescriptor& descriptor, |