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/workloads/Decoders.hpp

diff --git a/src/backends/reference/workloads/Decoders.hpp b/src/backends/reference/workloads/Decoders.hpp
index 83c57c1..3434ccb 100644
--- a/src/backends/reference/workloads/Decoders.hpp
+++ b/src/backends/reference/workloads/Decoders.hpp
@@ -136,6 +136,10 @@
                     info.GetQuantizationOffset());
             }
         }
+        case armnn::DataType::Boolean:
+        {
+            return std::make_unique<BooleanDecoder>(static_cast<const uint8_t*>(data));
+        }
         default:
         {
             BOOST_ASSERT_MSG(false, "Unsupported Data Type!");