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/cl/workloads/ClGreaterWorkload.cpp | |
| 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/cl/workloads/ClGreaterWorkload.cpp')
| -rw-r--r-- | src/backends/cl/workloads/ClGreaterWorkload.cpp | 12 |
1 files changed, 3 insertions, 9 deletions
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 |