commit ea54a01f6bd30f013cbe88ae1751985bc86b6af5
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Mon Mar 16 16:36:10 2020 +0000
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Mar 19 15:41:12 2020 +0000
tree 7edb7d659ea4210c1256beb5edf57601b317c82d
parent 25334cf3d53fe7fff98776b44a199ca341f62f1a

IVGCVSW-4516 Add ConvertFp32ToBf16Layer and Ref workload support

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

src/backends/backendsCommon/Workload.hpp

diff --git a/src/backends/backendsCommon/Workload.hpp b/src/backends/backendsCommon/Workload.hpp
index 56736ce..984443b 100644
--- a/src/backends/backendsCommon/Workload.hpp
+++ b/src/backends/backendsCommon/Workload.hpp
@@ -182,6 +182,11 @@
                                                      armnn::DataType::Float32>;
 
 template <typename QueueDescriptor>
+using Float32ToBFloat16Workload = MultiTypedWorkload<QueueDescriptor,
+                                                     armnn::DataType::Float32,
+                                                     armnn::DataType::BFloat16>;
+
+template <typename QueueDescriptor>
 using Float16ToFloat32Workload = MultiTypedWorkload<QueueDescriptor,
                                                     armnn::DataType::Float16,
                                                     armnn::DataType::Float32>;