diff options
Diffstat (limited to 'src/backends/backendsCommon/test/WorkloadTestUtils.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/WorkloadTestUtils.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp index 51683335e1..df001b7530 100644 --- a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp +++ b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp @@ -95,6 +95,7 @@ inline armnn::Optional<armnn::DataType> GetBiasTypeFromWeightsType(armnn::Option switch(weightsType.value()) { + case armnn::DataType::BFloat16: case armnn::DataType::Float16: case armnn::DataType::Float32: return weightsType; |