commit b60dd243d3d8131f246f2b122309b998287151d2
author Sadik Armagan <sadik.armagan@arm.com> Thu Mar 19 13:53:16 2020 +0000
committer Jim Flynn <jim.flynn@arm.com> Thu Mar 19 21:47:24 2020 +0000
tree af5e84f753236078006c32957a7d35f4ac708563
parent a0687eef149fbf57bb6db0621ec65724f550b1ed

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

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 9f22b9e..7d3600c 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -302,14 +302,15 @@
                                             Optional<std::string&> reasonIfUnsupported) const
 {
     IgnoreUnused(descriptor);
-
-    std::array<DataType, 5> supportedInputTypes =
+    std::array<DataType, 7> supportedInputTypes =
     {
+        DataType::Boolean,
         DataType::BFloat16,
         DataType::Float32,
         DataType::Float16,
         DataType::QAsymmU8,
-        DataType::QSymmS16
+        DataType::QSymmS16,
+        DataType::Signed32
     };
 
     bool supported = true;