commit 20e984f442483493ece882c3040785368eb31c96
author ruoyan01 <ruomei.yan@arm.com> Wed Dec 12 18:11:25 2018 +0000
committer ruoyan01 <ruomei.yan@arm.com> Thu Dec 13 12:51:23 2018 +0000
tree 33a9e50c955c31a3942b0e7806a7efdbfd8358d6
parent 482ca850aa980f9ed740fdb7537e2b74ddeb7561

IVGCVSW-2348 Support boolean data type

Change-Id: Ifd28e049192e6f5fe5c0f5d358afb2b530eef882

include/armnn/Types.hpp

diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp
index d815005..baf7443 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -26,9 +26,10 @@
 enum class DataType
 {
     Float16 = 0,
-    Float32   = 1,
+    Float32 = 1,
     QuantisedAsymm8 = 2,
-    Signed32  = 3
+    Signed32 = 3,
+    Boolean = 4
 };
 
 enum class DataLayout

include/armnn/TypesUtils.hpp

diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp
index 68ad455..7eacc00 100644
--- a/include/armnn/TypesUtils.hpp
+++ b/include/armnn/TypesUtils.hpp
@@ -79,11 +79,12 @@
 {
     switch (dataType)
     {
-        case DataType::Float16:     return 2U;
+        case DataType::Float16:          return 2U;
         case DataType::Float32:
-        case DataType::Signed32:   return 4U;
-        case DataType::QuantisedAsymm8: return 1U;
-        default:                  return 0U;
+        case DataType::Signed32:         return 4U;
+        case DataType::QuantisedAsymm8:  return 1U;
+        case DataType::Boolean:          return 1U;
+        default:                         return 0U;
     }
 }
 
@@ -167,6 +168,12 @@
     static constexpr DataType Value = DataType::Signed32;
 };
 
+template<>
+struct GetDataTypeImpl<bool>
+{
+    static constexpr DataType Value = DataType::Boolean;
+};
+
 template <typename T>
 constexpr DataType GetDataType()
 {