aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonConstantWorkload.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/workloads/NeonConstantWorkload.cpp')
-rw-r--r--src/backends/neon/workloads/NeonConstantWorkload.cpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonConstantWorkload.cpp b/src/backends/neon/workloads/NeonConstantWorkload.cpp
index 08f0390e34..83a2692b6e 100644
--- a/src/backends/neon/workloads/NeonConstantWorkload.cpp
+++ b/src/backends/neon/workloads/NeonConstantWorkload.cpp
@@ -6,6 +6,7 @@
#include "NeonConstantWorkload.hpp"
#include <arm_compute/core/Types.h>
+#include <BFloat16.hpp>
#include <Half.hpp>
#include <aclCommon/ArmComputeTensorUtils.hpp>
#include <neon/NeonTensorHandle.hpp>
@@ -46,6 +47,11 @@ void NeonConstantWorkload::Execute() const
switch (computeDataType)
{
+ case arm_compute::DataType::BFLOAT16:
+ {
+ CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<BFloat16>(), output);
+ break;
+ }
case arm_compute::DataType::F16:
{
CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<Half>(), output);