diff options
Diffstat (limited to 'src/backends/reference/workloads/RefTransposeWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefTransposeWorkload.cpp | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefTransposeWorkload.cpp b/src/backends/reference/workloads/RefTransposeWorkload.cpp index cc7a555c41..828badd042 100644 --- a/src/backends/reference/workloads/RefTransposeWorkload.cpp +++ b/src/backends/reference/workloads/RefTransposeWorkload.cpp @@ -16,12 +16,25 @@ namespace armnn template <armnn::DataType DataType> void RefTransposeWorkload<DataType>::Execute() const { + Execute(m_Data.m_Inputs, m_Data.m_Outputs); +} + +template <armnn::DataType DataType> +void RefTransposeWorkload<DataType>::ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) +{ + Execute(workingMemDescriptor.m_Inputs, workingMemDescriptor.m_Outputs); +} + +template <armnn::DataType DataType> +void RefTransposeWorkload<DataType>::Execute(std::vector<ITensorHandle*> inputs, + std::vector<ITensorHandle*> outputs) const +{ using T = ResolveType<DataType>; ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, GetName() + "_Execute"); - const ITensorHandle* src = m_Data.m_Inputs[0]; - ITensorHandle* dst = m_Data.m_Outputs[0]; + const ITensorHandle* src = inputs[0]; + ITensorHandle* dst = outputs[0]; const PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; armnnUtils::Transpose(GetTensorInfo(src).GetShape(), mappings, src->Map(), dst->Map(), sizeof(T)); |