// // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once #include #include namespace armnn { arm_compute::Status NeonGreaterWorkloadValidate(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& output); template class NeonGreaterWorkload : public MultiTypedWorkload { public: using MultiTypedWorkload::m_Data; NeonGreaterWorkload(const GreaterQueueDescriptor& descriptor, const WorkloadInfo& info); virtual void Execute() const override; private: mutable arm_compute::NEGreater m_GreaterLayer; }; using NeonGreaterFloat32Workload = NeonGreaterWorkload; using NeonGreaterUint8Workload = NeonGreaterWorkload; } //namespace armnn