enum class DataType
{
Float16 = 0,
- Float32 = 1,
+ Float32 = 1,
QuantisedAsymm8 = 2,
- Signed32 = 3
+ Signed32 = 3,
+ Boolean = 4
};
enum class DataLayout
{
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;
}
}
static constexpr DataType Value = DataType::Signed32;
};
+template<>
+struct GetDataTypeImpl<bool>
+{
+ static constexpr DataType Value = DataType::Boolean;
+};
+
template <typename T>
constexpr DataType GetDataType()
{
using Type = float;
};
+template<>
+struct ResolveTypeImpl<DataType::Boolean>
+{
+ using Type = bool;
+};
+
template<DataType DT>
using ResolveType = typename ResolveTypeImpl<DT>::Type;
BOOST_TEST(armnn::GetDataTypeSize(armnn::DataType::Float32) == 4);
BOOST_TEST(armnn::GetDataTypeSize(armnn::DataType::QuantisedAsymm8) == 1);
BOOST_TEST(armnn::GetDataTypeSize(armnn::DataType::Signed32) == 4);
+ BOOST_TEST(armnn::GetDataTypeSize(armnn::DataType::Boolean) == 1);
}
BOOST_AUTO_TEST_CASE(GetDataTypeTest)
BOOST_TEST((armnn::GetDataType<float>() == armnn::DataType::Float32));
BOOST_TEST((armnn::GetDataType<uint8_t>() == armnn::DataType::QuantisedAsymm8));
BOOST_TEST((armnn::GetDataType<int32_t>() == armnn::DataType::Signed32));
+ BOOST_TEST((armnn::GetDataType<bool>() == armnn::DataType::Boolean));
}
BOOST_AUTO_TEST_CASE(PermuteDescriptorWithTooManyMappings)