diff options
Diffstat (limited to 'src/backends/backendsCommon/Workload.hpp')
| -rw-r--r-- | src/backends/backendsCommon/Workload.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/Workload.hpp b/src/backends/backendsCommon/Workload.hpp index 34d13635ba..4d14adbf54 100644 --- a/src/backends/backendsCommon/Workload.hpp +++ b/src/backends/backendsCommon/Workload.hpp @@ -165,6 +165,19 @@ template <typename QueueDescriptor> using Int32Workload = TypedWorkload<QueueDescriptor, armnn::DataType::Signed32>; template <typename QueueDescriptor> +using BooleanWorkload = TypedWorkload<QueueDescriptor, armnn::DataType::Boolean>; + +template <typename QueueDescriptor> +using BaseFloat32ComparisonWorkload = MultiTypedWorkload<QueueDescriptor, + armnn::DataType::Float32, + armnn::DataType::Boolean>; + +template <typename QueueDescriptor> +using BaseUint8ComparisonWorkload = MultiTypedWorkload<QueueDescriptor, + armnn::DataType::QuantisedAsymm8, + armnn::DataType::Boolean>; + +template <typename QueueDescriptor> using Float16ToFloat32Workload = MultiTypedWorkload<QueueDescriptor, armnn::DataType::Float16, armnn::DataType::Float32>; |