diff options
Diffstat (limited to 'src/backends/cl')
-rw-r--r-- | src/backends/cl/ClWorkloadFactory.cpp | 3 | ||||
-rw-r--r-- | src/backends/cl/workloads/ClGreaterWorkload.cpp | 12 | ||||
-rw-r--r-- | src/backends/cl/workloads/ClGreaterWorkload.hpp | 7 |
3 files changed, 5 insertions, 17 deletions
diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp index e7e4fa7e1b..4f707beebe 100644 --- a/src/backends/cl/ClWorkloadFactory.cpp +++ b/src/backends/cl/ClWorkloadFactory.cpp @@ -178,8 +178,7 @@ std::unique_ptr<IWorkload> ClWorkloadFactory::CreateComparison(const ComparisonQ GreaterQueueDescriptor greaterQueueDescriptor; greaterQueueDescriptor.m_Inputs = descriptor.m_Inputs; greaterQueueDescriptor.m_Outputs = descriptor.m_Outputs; - - return MakeWorkload<ClGreaterFloat32Workload, ClGreaterUint8Workload>(greaterQueueDescriptor, info); + return MakeWorkload<ClGreaterWorkload>(greaterQueueDescriptor, info); } return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info); } diff --git a/src/backends/cl/workloads/ClGreaterWorkload.cpp b/src/backends/cl/workloads/ClGreaterWorkload.cpp index b086122bdc..2051cc3aa3 100644 --- a/src/backends/cl/workloads/ClGreaterWorkload.cpp +++ b/src/backends/cl/workloads/ClGreaterWorkload.cpp @@ -38,10 +38,8 @@ arm_compute::Status ClGreaterWorkloadValidate(const TensorInfo& input0, return aclStatus; } -template<DataType T> -ClGreaterWorkload<T>::ClGreaterWorkload(const GreaterQueueDescriptor& descriptor, - const WorkloadInfo& info) - : MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>(descriptor, info) +ClGreaterWorkload::ClGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info) + : BaseWorkload<GreaterQueueDescriptor>(descriptor, info) { m_Data.ValidateInputsOutputs("ClGreaterWorkload", 2, 1); @@ -52,14 +50,10 @@ ClGreaterWorkload<T>::ClGreaterWorkload(const GreaterQueueDescriptor& descriptor m_GreaterLayer.configure(&input0, &input1, &output, arm_compute::ComparisonOperation::Greater); } -template<DataType T> -void ClGreaterWorkload<T>::Execute() const +void ClGreaterWorkload::Execute() const { ARMNN_SCOPED_PROFILING_EVENT_CL("ClGreaterWorkload_Execute"); RunClFunction(m_GreaterLayer, CHECK_LOCATION()); } -template class ClGreaterWorkload<DataType::Float32>; -template class ClGreaterWorkload<DataType::QAsymmU8>; - } //namespace armnn 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 |