diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-03-19 13:53:16 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-03-19 21:47:24 +0000 |
commit | b60dd243d3d8131f246f2b122309b998287151d2 (patch) | |
tree | af5e84f753236078006c32957a7d35f4ac708563 /src/backends/neon/workloads/NeonGreaterWorkload.hpp | |
parent | a0687eef149fbf57bb6db0621ec65724f550b1ed (diff) | |
download | armnn-b60dd243d3d8131f246f2b122309b998287151d2.tar.gz |
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 <sadik.armagan@arm.com>
Change-Id: I92772810b744b388831c9dca0119ebf8cb7a447e
Diffstat (limited to 'src/backends/neon/workloads/NeonGreaterWorkload.hpp')
-rw-r--r-- | src/backends/neon/workloads/NeonGreaterWorkload.hpp | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/src/backends/neon/workloads/NeonGreaterWorkload.hpp b/src/backends/neon/workloads/NeonGreaterWorkload.hpp index bcab27e7a6..503e60e784 100644 --- a/src/backends/neon/workloads/NeonGreaterWorkload.hpp +++ b/src/backends/neon/workloads/NeonGreaterWorkload.hpp @@ -16,12 +16,9 @@ arm_compute::Status NeonGreaterWorkloadValidate(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& output); -template <DataType T> -class NeonGreaterWorkload : public MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean> +class NeonGreaterWorkload : public BaseWorkload<GreaterQueueDescriptor> { public: - using MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>::m_Data; - NeonGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info); virtual void Execute() const override; @@ -30,7 +27,4 @@ private: mutable arm_compute::NEGreater m_GreaterLayer; }; -using NeonGreaterFloat32Workload = NeonGreaterWorkload<DataType::Float32>; -using NeonGreaterUint8Workload = NeonGreaterWorkload<DataType::QAsymmU8>; - } //namespace armnn
\ No newline at end of file |