diff options
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseWorkload.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefElementwiseWorkload.hpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.hpp b/src/backends/reference/workloads/RefElementwiseWorkload.hpp index 2772b77631..b5205938b2 100644 --- a/src/backends/reference/workloads/RefElementwiseWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseWorkload.hpp @@ -10,6 +10,9 @@ #include <backendsCommon/Workload.hpp> #include <backendsCommon/WorkloadData.hpp> #include "Maximum.hpp" +#include "Minimum.hpp" + + namespace armnn { @@ -133,4 +136,15 @@ using RefMaximumUint8Workload = MaximumQueueDescriptor, StringMapping::RefMaximumWorkload_Execute>; +using RefMinimumFloat32Workload = + RefElementwiseWorkload<minimum<float>, + DataType::Float32, + MinimumQueueDescriptor, + StringMapping::RefMinimumWorkload_Execute>; + +using RefMinimumUint8Workload = + RefElementwiseWorkload<minimum<float>, + DataType::QuantisedAsymm8, + MinimumQueueDescriptor, + StringMapping::RefMinimumWorkload_Execute>; } // armnn |