1 // 2 // Copyright © 2020 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <backendsCommon/Workload.hpp> 9 #include <backendsCommon/WorkloadData.hpp> 10 #include <neon/workloads/NeonWorkloadUtils.hpp> 11 12 namespace armnn 13 { 14 15 class NeonConvertBf16ToFp32Workload : public BFloat16ToFloat32Workload<ConvertBf16ToFp32QueueDescriptor> 16 { 17 public: 18 NeonConvertBf16ToFp32Workload(const ConvertBf16ToFp32QueueDescriptor& descriptor, const WorkloadInfo& info); 19 virtual void Execute() const override; 20 21 private: 22 using TensorHandlePair = std::pair<const ITensorHandle*, ITensorHandle*>; 23 std::vector<TensorHandlePair> m_TensorHandlePairs; 24 }; 25 26 } //namespace armnn 27