diff options
Diffstat (limited to 'src/armnnUtils/QuantizeHelper.hpp')
-rw-r--r-- | src/armnnUtils/QuantizeHelper.hpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnnUtils/QuantizeHelper.hpp b/src/armnnUtils/QuantizeHelper.hpp index 6fd13fda98..596ec98f64 100644 --- a/src/armnnUtils/QuantizeHelper.hpp +++ b/src/armnnUtils/QuantizeHelper.hpp @@ -8,6 +8,7 @@ #include <armnn/utility/IgnoreUnused.hpp> #include <armnn/TypesUtils.hpp> +#include <BFloat16.hpp> #include <Half.hpp> #include <initializer_list> @@ -65,6 +66,22 @@ struct SelectiveQuantizer<armnn::Half, false> } }; +template<> +struct SelectiveQuantizer<armnn::BFloat16, false> +{ + static armnn::BFloat16 Quantize(float value, float scale, int32_t offset) + { + armnn::IgnoreUnused(scale, offset); + return armnn::BFloat16(value); + } + + static float Dequantize(armnn::BFloat16 value, float scale, int32_t offset) + { + armnn::IgnoreUnused(scale, offset); + return value; + } +}; + template<typename T> T SelectiveQuantize(float value, float scale, int32_t offset) { |