diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonConstantWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonConstantWorkload.cpp | 6 |
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); |