aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefStridedSliceWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefStridedSliceWorkload.hpp')
-rw-r--r--src/backends/reference/workloads/RefStridedSliceWorkload.hpp20
1 files changed, 3 insertions, 17 deletions
diff --git a/src/backends/reference/workloads/RefStridedSliceWorkload.hpp b/src/backends/reference/workloads/RefStridedSliceWorkload.hpp
index b3586adbda..44aabc0106 100644
--- a/src/backends/reference/workloads/RefStridedSliceWorkload.hpp
+++ b/src/backends/reference/workloads/RefStridedSliceWorkload.hpp
@@ -7,28 +7,14 @@
#include <backendsCommon/Workload.hpp>
-#include <armnn/TypesUtils.hpp>
-
namespace armnn
{
-template <armnn::DataType DataType>
-class RefStridedSliceWorkload : public TypedWorkload<StridedSliceQueueDescriptor, DataType>
+class RefStridedSliceWorkload : public BaseWorkload<StridedSliceQueueDescriptor>
{
public:
- static const std::string& GetName()
- {
- static const std::string name = std::string("RefStridedSlice") + GetDataTypeName(DataType) + "Workload";
- return name;
- }
-
- using TypedWorkload<StridedSliceQueueDescriptor, DataType>::m_Data;
- using TypedWorkload<StridedSliceQueueDescriptor, DataType>::TypedWorkload;
-
+ RefStridedSliceWorkload(const StridedSliceQueueDescriptor& descriptor, const WorkloadInfo& info);
void Execute() const override;
};
-using RefStridedSliceFloat32Workload = RefStridedSliceWorkload<DataType::Float32>;
-using RefStridedSliceUint8Workload = RefStridedSliceWorkload<DataType::QuantisedAsymm8>;
-
-} //namespace armnn
+} // namespace armnn