diff options
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseWorkload.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefElementwiseWorkload.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.hpp b/src/backends/reference/workloads/RefElementwiseWorkload.hpp index b5205938b2..c2855b0550 100644 --- a/src/backends/reference/workloads/RefElementwiseWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseWorkload.hpp @@ -147,4 +147,16 @@ using RefMinimumUint8Workload = DataType::QuantisedAsymm8, MinimumQueueDescriptor, StringMapping::RefMinimumWorkload_Execute>; + +using RefEqualFloat32Workload = + RefElementwiseWorkload<std::equal_to<float>, + DataType::Float32, + EqualQueueDescriptor, + StringMapping::RefEqualWorkload_Execute>; + +using RefEqualUint8Workload = + RefElementwiseWorkload<std::equal_to<float>, + DataType::QuantisedAsymm8, + EqualQueueDescriptor, + StringMapping::RefEqualWorkload_Execute>; } // armnn |