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 /src/armnn/ResolveType.hpp | |
| 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 'src/armnn/ResolveType.hpp')
| -rw-r--r-- | src/armnn/ResolveType.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/ResolveType.hpp b/src/armnn/ResolveType.hpp index 66309344db..e1bea42d3c 100644 --- a/src/armnn/ResolveType.hpp +++ b/src/armnn/ResolveType.hpp @@ -6,6 +6,7 @@ #pragma once #include "armnn/Types.hpp" +#include "BFloat16.hpp" #include "Half.hpp" namespace armnn @@ -62,6 +63,12 @@ struct ResolveTypeImpl<DataType::Boolean> using Type = uint8_t; }; +template<> +struct ResolveTypeImpl<DataType::BFloat16> +{ + using Type = BFloat16; +}; + template<DataType DT> using ResolveType = typename ResolveTypeImpl<DT>::Type; |