diff options
Diffstat (limited to 'src/backends/cl/workloads/ClGreaterWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClGreaterWorkload.hpp | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/src/backends/cl/workloads/ClGreaterWorkload.hpp b/src/backends/cl/workloads/ClGreaterWorkload.hpp index 862e168334..9b2a1710bc 100644 --- a/src/backends/cl/workloads/ClGreaterWorkload.hpp +++ b/src/backends/cl/workloads/ClGreaterWorkload.hpp @@ -16,19 +16,14 @@ arm_compute::Status ClGreaterWorkloadValidate(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& output); -template<DataType T> -class ClGreaterWorkload : public MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean> +class ClGreaterWorkload : public BaseWorkload<GreaterQueueDescriptor> { public: ClGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info); void Execute() const override; private: - using MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>::m_Data; mutable arm_compute::CLComparison m_GreaterLayer; }; -using ClGreaterFloat32Workload = ClGreaterWorkload<DataType::Float32>; -using ClGreaterUint8Workload = ClGreaterWorkload<DataType::QAsymmU8>; - } //namespace armnn |