diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-02-28 12:45:21 +0000 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-03-03 15:59:39 +0000 |
| commit | c3bf6efb48a4540c8addcc02813c9381e4fceb1f (patch) | |
| tree | 55fd9c3fa639ee32aa847dda2a525a7badee9e8d /include | |
| parent | 490b7becb8029ead26423b0d62e631a929e55d6c (diff) | |
| download | armnn-c3bf6efb48a4540c8addcc02813c9381e4fceb1f.tar.gz | |
IVGCVSW-4508 Add BFloat16 data type
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie2fcf06d0bae9e2ef958e60ab9e8b121fdc7b590
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/Types.hpp | 1 | ||||
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 2 |
2 files changed, 3 insertions, 0 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index d66a1fda90..29a0d4e364 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -40,6 +40,7 @@ enum class DataType QuantizedSymm8PerAxis ARMNN_DEPRECATED_ENUM_MSG("Per Axis property inferred by number of scales in TensorInfo") = 6, QSymmS8 = 7, QAsymmS8 = 8, + BFloat16 = 9, QuantisedAsymm8 ARMNN_DEPRECATED_ENUM_MSG("Use DataType::QAsymmU8 instead.") = QAsymmU8, QuantisedSymm16 ARMNN_DEPRECATED_ENUM_MSG("Use DataType::QSymmS16 instead.") = QSymmS16 diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 53a933b480..5065152c7a 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -116,6 +116,7 @@ constexpr unsigned int GetDataTypeSize(DataType dataType) { switch (dataType) { + case DataType::BFloat16: case DataType::Float16: return 2U; case DataType::Float32: case DataType::Signed32: return 4U; @@ -179,6 +180,7 @@ constexpr const char* GetDataTypeName(DataType dataType) case DataType::QSymmS16: return "QSymm16"; case DataType::Signed32: return "Signed32"; case DataType::Boolean: return "Boolean"; + case DataType::BFloat16: return "BFloat16"; default: return "Unknown"; |