| // |
| // Copyright © 2017 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #pragma once |
| |
| #include <armnn/Types.hpp> |
| #include <backendsCommon/Workload.hpp> |
| #include <backendsCommon/WorkloadData.hpp> |
| #include "StringMapping.hpp" |
| |
| namespace armnn |
| { |
| |
| template <typename Functor, typename ParentDescriptor, typename armnn::StringMapping::Id DebugString> |
| class RefComparisonWorkload : public BaseWorkload<ParentDescriptor> |
| { |
| public: |
| using BaseWorkload<ParentDescriptor>::m_Data; |
| using BaseWorkload<ParentDescriptor>::BaseWorkload; |
| |
| void Execute() const override; |
| }; |
| |
| using RefEqualWorkload = |
| RefComparisonWorkload<std::equal_to<float>, |
| EqualQueueDescriptor, |
| StringMapping::RefEqualWorkload_Execute>; |
| |
| |
| using RefGreaterWorkload = |
| RefComparisonWorkload<std::greater<float>, |
| GreaterQueueDescriptor, |
| StringMapping::RefGreaterWorkload_Execute>; |
| } // armnn |