From a856501ac5c2d0cca70068993e5c7cc714872890 Mon Sep 17 00:00:00 2001 From: Keith Davis Date: Fri, 14 Feb 2020 12:22:40 +0000 Subject: IVGCVSW-4420 Failing Unit Test Cl: Quantize layer * Fix bug with missing ValidateInputsOutputs in Cl Workload * Renamed data type tests to be more specific * Future proofing Signed-off-by: Keith Davis Change-Id: I157acc318e40d727fcfb3f36f7f577fc87472f3d --- src/backends/cl/workloads/ClQuantizeWorkload.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/backends/cl/workloads/ClQuantizeWorkload.cpp') diff --git a/src/backends/cl/workloads/ClQuantizeWorkload.cpp b/src/backends/cl/workloads/ClQuantizeWorkload.cpp index 230e346a00..263065a5a4 100644 --- a/src/backends/cl/workloads/ClQuantizeWorkload.cpp +++ b/src/backends/cl/workloads/ClQuantizeWorkload.cpp @@ -32,6 +32,8 @@ arm_compute::Status ClQuantizeWorkloadValidate(const TensorInfo& input, ClQuantizeWorkload::ClQuantizeWorkload(const QuantizeQueueDescriptor& descriptor, const WorkloadInfo& info) : BaseWorkload(descriptor, info) { + m_Data.ValidateInputsOutputs("ClQuantizeWorkload", 1, 1); + arm_compute::ICLTensor& input = static_cast(m_Data.m_Inputs[0])->GetTensor(); arm_compute::ICLTensor& output = static_cast(m_Data.m_Outputs[0])->GetTensor(); -- cgit v1.2.1