src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.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 "NeonConvolution2dFloat32Workload.hpp"
   7 #include "backends/CpuTensorHandle.hpp"
   8 #include "backends/ArmComputeTensorUtils.hpp"
   9 #include "backends/NeonLayerSupport.hpp"
  10
  11 namespace armnn
  12 {
  13 using namespace armcomputetensorutils;
  14
  15 NeonConvolution2dFloat32Workload::NeonConvolution2dFloat32Workload(const Convolution2dQueueDescriptor& descriptor,
  16                                                                    const WorkloadInfo& info)
  17     : NeonConvolution2dBaseWorkload(descriptor, info)
  18 {
  19     if (m_Data.m_Parameters.m_BiasEnabled)
  20     {
  21         InitialiseArmComputeTensorData(m_BiasTensor, m_Data.m_Bias->template GetConstTensor<float>());
  22     }
  23 }
  24
  25
  26 void NeonConvolution2dFloat32Workload::Execute() const
  27 {
  28     ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuAcc, "NeonConvolution2dFloat32Workload_Execute");
  29     m_ConvolutionLayer->run();
  30 }
  31
  32 void NeonConvolution2dFloat32Workload::ValidateData() const
  33 {
  34     m_Data.ValidateInputsOutputs("NeonConvolution2dFloat32Workload", 1, 1);
  35 }
  36
  37
  38
  39 } //namespace armnn
  40
  41