commit 7ddbbae7ad3e0000d8e6a76458cac68254dc8048
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Mar 13 10:26:05 2020 +0000
committer Jim Flynn <jim.flynn@arm.com> Tue Mar 17 20:56:46 2020 +0000
tree 43f6240df090b084528034358982e8f09706ef95
parent f4a953f75b751452ae9303abc8565d310c55bfff

IVGCVSW-4515 Add ConvertBf16ToFp32Layer and Ref workload support

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ida6d7e1d2c9abe0618f8b711bab9d62c011090d6

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 94128fe..a4f4efd 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -376,6 +376,21 @@
                                   "Reference constant: output is not a supported type.");
 }
 
+bool RefLayerSupport::IsConvertBf16ToFp32Supported(const TensorInfo& input,
+                                                   const TensorInfo& output,
+                                                   Optional<std::string&> reasonIfUnsupported) const
+{
+    bool supported = true;
+
+    supported &= CheckSupportRule(TypeIs(input, DataType::BFloat16), reasonIfUnsupported,
+                                  "Reference for ConvertBf16ToFp32 layer: input type not supported");
+
+    supported &= CheckSupportRule(TypeIs(output, DataType::Float32), reasonIfUnsupported,
+                                  "Reference for ConvertBf16ToFp32 layer: output type not supported");
+
+    return supported;
+}
+
 bool RefLayerSupport::IsConvertFp16ToFp32Supported(const TensorInfo& input,
                                                    const TensorInfo& output,
                                                    Optional<std::string&> reasonIfUnsupported) const