aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/TypesUtils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r--include/armnn/TypesUtils.hpp15
1 files changed, 11 insertions, 4 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp
index 68ad45546d..7eacc00a93 100644
--- a/include/armnn/TypesUtils.hpp
+++ b/include/armnn/TypesUtils.hpp
@@ -79,11 +79,12 @@ constexpr unsigned int GetDataTypeSize(DataType dataType)
{
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 @@ struct GetDataTypeImpl<int32_t>
static constexpr DataType Value = DataType::Signed32;
};
+template<>
+struct GetDataTypeImpl<bool>
+{
+ static constexpr DataType Value = DataType::Boolean;
+};
+
template <typename T>
constexpr DataType GetDataType()
{