src/armnn/backends/NeonWorkloads/NeonPermuteWorkload.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5
   6 #include "NeonPermuteWorkload.hpp"
   7 #include "backends/NeonTensorHandle.hpp"
   8 #include "backends/ArmComputeTensorUtils.hpp"
   9
  10 #include <arm_compute/core/Error.h>
  11
  12 namespace armnn
  13 {
  14
  15 arm_compute::Status NeonPermuteWorkloadValidate(const TensorInfo& input,
  16                                                 const TensorInfo& output,
  17                                                 const PermuteDescriptor& descriptor)
  18 {
  19     const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input);
  20     const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
  21     const armnn::PermutationVector& mappings = descriptor.m_DimMappings;
  22
  23     return arm_compute::NEPermute::validate(&aclInputInfo, &aclOutputInfo,
  24                                       armcomputetensorutils::BuildArmComputePermutationVector(mappings));
  25 }
  26
  27 template <armnn::DataType... DataTypes>
  28 NeonPermuteWorkload<DataTypes...>::NeonPermuteWorkload(const PermuteQueueDescriptor& descriptor,
  29                                                const WorkloadInfo& info)
  30         : TypedWorkload<PermuteQueueDescriptor, DataTypes...>(descriptor, info)
  31 {
  32     using armcomputetensorutils::BuildArmComputePermutationVector;
  33
  34     m_Data.ValidateInputsOutputs(GetName(), 1, 1);
  35
  36     const arm_compute::ITensor& input = static_cast<INeonTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
  37     arm_compute::ITensor& output = static_cast<INeonTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
  38     const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings;
  39
  40     // Run the layer.
  41     m_PermuteFunction.configure(&input, &output, BuildArmComputePermutationVector(mappings));
  42 }
  43
  44 template <armnn::DataType... DataTypes>
  45 void NeonPermuteWorkload<DataTypes...>::Execute() const
  46 {
  47     ARMNN_SCOPED_PROFILING_EVENT_NEON(GetName() + "_Execute");
  48     m_PermuteFunction.run();
  49 }
  50
  51 template class NeonPermuteWorkload<DataType::Float16, DataType::Float32>;
  52 template class NeonPermuteWorkload<DataType::QuantisedAsymm8>;
  53
  54 } // namespace armnn