From 44179c372eea9f17c96cbf50ee383e57e14d70a6 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 11 Mar 2020 14:51:27 +0000 Subject: IVGCVSW-4511 Add BFloat16 to RefLayerSupport and unit tests Signed-off-by: Narumol Prangnawarat Change-Id: Ifaae4d5aac468ba927b2c6a4bf31b8c8522aeb2e --- src/backends/backendsCommon/WorkloadFactory.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp') diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index 6ac76ecea6..2e1ce0a674 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -325,6 +325,7 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId, TensorInfo biasInfo; const TensorInfo * biasInfoPtr = nullptr; + static const TensorInfo dummyBFloat16Bias(TensorShape({1,1,1,1}), DataType::BFloat16); static const TensorInfo dummyFloat16Bias(TensorShape({1,1,1,1}), DataType::Float16); static const TensorInfo dummyFloat32Bias(TensorShape({1,1,1,1}), DataType::Float32); static const TensorInfo dummyQA8Bias(TensorShape({1,1,1,1}), DataType::Signed32); @@ -341,6 +342,11 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId, // If biases are not enabled pass a dummy tensorinfo for the validation switch(input.GetDataType()) { + case DataType::BFloat16: + { + biasInfoPtr = &dummyBFloat16Bias; + break; + } case DataType::Float16: { biasInfoPtr = &dummyFloat16Bias; -- cgit v1.2.1