1 // 2 // Copyright © 2020 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "RefTransposeWorkload.hpp" 7 #include "RefWorkloadUtils.hpp" 8 9 #include <armnnUtils/Transpose.hpp> 10 11 #include <ResolveType.hpp> 12 13 namespace armnn 14 { 15 16 template <armnn::DataType DataType> Execute() const17void RefTransposeWorkload<DataType>::Execute() const 18 { 19 using T = ResolveType<DataType>; 20 21 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, GetName() + "_Execute"); 22 23 const ITensorHandle* src = m_Data.m_Inputs[0]; 24 ITensorHandle* dst = m_Data.m_Outputs[0]; 25 const PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; 26 27 armnnUtils::Transpose(GetTensorInfo(src).GetShape(), mappings, src->Map(), dst->Map(), sizeof(T)); 28 } 29 30 template class RefTransposeWorkload<DataType::BFloat16>; 31 template class RefTransposeWorkload<DataType::Float16>; 32 template class RefTransposeWorkload<DataType::Float32>; 33 template class RefTransposeWorkload<DataType::QAsymmS8>; 34 template class RefTransposeWorkload<DataType::QAsymmU8>; 35 template class RefTransposeWorkload<DataType::QSymmS16>; 36 37 } //namespace armnn 38