diff options
Diffstat (limited to 'src/armnn/CompatibleTypes.hpp')
-rw-r--r-- | src/armnn/CompatibleTypes.hpp | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/armnn/CompatibleTypes.hpp b/src/armnn/CompatibleTypes.hpp index 4332f74b23..1a663d3e27 100644 --- a/src/armnn/CompatibleTypes.hpp +++ b/src/armnn/CompatibleTypes.hpp @@ -5,8 +5,10 @@ #pragma once -#include "armnn/Types.hpp" -#include "Half.hpp" +#include <armnn/Types.hpp> + +#include <BFloat16.hpp> +#include <Half.hpp> namespace armnn { @@ -30,6 +32,12 @@ inline bool CompatibleTypes<Half>(DataType dataType) } template<> +inline bool CompatibleTypes<BFloat16>(DataType dataType) +{ + return dataType == DataType::BFloat16; +} + +template<> inline bool CompatibleTypes<uint8_t>(DataType dataType) { return dataType == DataType::Boolean || dataType == DataType::QAsymmU8; |