src/backends/neon/workloads/NeonPermuteWorkload.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "NeonPermuteWorkload.hpp"
   7 #include <neon/NeonTensorHandle.hpp>
   8 #include <aclCommon/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 NeonPermuteWorkload::NeonPermuteWorkload(const PermuteQueueDescriptor& descriptor,
  28                                          const WorkloadInfo& info)
  29         : BaseWorkload<PermuteQueueDescriptor>(descriptor, info)
  30 {
  31     using armcomputetensorutils::BuildArmComputePermutationVector;
  32
  33     m_Data.ValidateInputsOutputs(GetName(), 1, 1);
  34
  35     const arm_compute::ITensor& input = static_cast<INeonTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
  36     arm_compute::ITensor& output = static_cast<INeonTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
  37     const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings;
  38
  39     // Run the layer.
  40     m_PermuteFunction.configure(&input, &output, BuildArmComputePermutationVector(mappings));
  41 }
  42
  43 void NeonPermuteWorkload::Execute() const
  44 {
  45     ARMNN_SCOPED_PROFILING_EVENT_NEON(GetName() + "_Execute");
  46     m_PermuteFunction.run();
  47 }
  48
  49 } // namespace armnn