From c3bf6efb48a4540c8addcc02813c9381e4fceb1f Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Fri, 28 Feb 2020 12:45:21 +0000
Subject: IVGCVSW-4508 Add BFloat16 data type

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie2fcf06d0bae9e2ef958e60ab9e8b121fdc7b590
---
 src/armnn/ResolveType.hpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/armnn/ResolveType.hpp')

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;
 
-- 
cgit v1.2.1