aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefStridedSliceWorkload.hpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-05-28 14:31:20 +0100
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-05-31 11:57:50 +0100
commite851b3da2ba51edc69c7b3dbfad06c4e22a63595 (patch)
tree5ead856b8c4de5198170f8ff3fdb2541eb6676d9 /src/backends/reference/workloads/RefStridedSliceWorkload.hpp
parent01961a7df1c4357981a33b9c1eb80fb51888a8fa (diff)
downloadarmnn-e851b3da2ba51edc69c7b3dbfad06c4e22a63595.tar.gz
IVGCVSW-3170 Refactor the Strided Slice Ref workload for Float32 and
QAsymm8 types * RefStridedSliceWorkload is no longer a template class * Refactoring of the ref StridedSlice implementation * Added ValidateTensorQuantizationSpace function Change-Id: Ifa182a33d79d42137731f48b995a7973c9d92152 Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
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