From b60dd243d3d8131f246f2b122309b998287151d2 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 19 Mar 2020 13:53:16 +0000 Subject: IVGCVSW-4565 TENSOR_BOOL8 data type not supported in AndroidNN Driver * Enabled Boolean and Int32 data types in Reference Comparison inputs * Added decoder for Boolean data type * Refactored ClGreaterWorkload to work with any data type * Refactored NeonGreaterWorkload to work with any data type !android-nn-driver:2902 Signed-off-by: Sadik Armagan Change-Id: I92772810b744b388831c9dca0119ebf8cb7a447e --- src/backends/cl/workloads/ClGreaterWorkload.hpp | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) (limited to 'src/backends/cl/workloads/ClGreaterWorkload.hpp') 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 -class ClGreaterWorkload : public MultiTypedWorkload +class ClGreaterWorkload : public BaseWorkload { public: ClGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info); void Execute() const override; private: - using MultiTypedWorkload::m_Data; mutable arm_compute::CLComparison m_GreaterLayer; }; -using ClGreaterFloat32Workload = ClGreaterWorkload; -using ClGreaterUint8Workload = ClGreaterWorkload; - } //namespace armnn -- cgit v1.2.1