src/backends/neon/workloads/NeonGreaterWorkload.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #pragma once
   7
   8 #include <neon/workloads/NeonWorkloadUtils.hpp>
   9
  10 #include <arm_compute/runtime/NEON/functions/NEElementwiseOperations.h>
  11
  12 namespace armnn
  13 {
  14
  15 arm_compute::Status NeonGreaterWorkloadValidate(const TensorInfo& input0,
  16                                                 const TensorInfo& input1,
  17                                                 const TensorInfo& output);
  18
  19 template <DataType T>
  20 class NeonGreaterWorkload : public MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>
  21 {
  22 public:
  23     using MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>::m_Data;
  24
  25     NeonGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info);
  26
  27     virtual void Execute() const override;
  28
  29 private:
  30     mutable arm_compute::NEGreater m_GreaterLayer;
  31 };
  32
  33 using NeonGreaterFloat32Workload = NeonGreaterWorkload<DataType::Float32>;
  34 using NeonGreaterUint8Workload = NeonGreaterWorkload<DataType::QAsymmU8>;
  35
  36 } //namespace armnn