From 5204aa8fd1da75ccc052269f358178b54bc7792f Mon Sep 17 00:00:00 2001
From: Keith Davis <keith.davis@arm.com>
Date: Mon, 27 Jan 2020 15:24:59 +0000
Subject: IVGCVSW-4306 Verify the tflite Yolo v3

 * Added debug layer support for QSymmS8
 * QSymmS8 support for workloads

Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I51af92fadc0be290629dd9198beab5abef9e351f
---
 src/backends/reference/workloads/Debug.cpp               | 6 ++++++
 src/backends/reference/workloads/RefDebugWorkload.cpp    | 1 +
 src/backends/reference/workloads/RefDebugWorkload.hpp    | 9 +++++----
 src/backends/reference/workloads/RefQuantizeWorkload.cpp | 2 +-
 4 files changed, 13 insertions(+), 5 deletions(-)

(limited to 'src/backends/reference/workloads')

diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp
index 0f192f3ab3..49e9e02ffb 100644
--- a/src/backends/reference/workloads/Debug.cpp
+++ b/src/backends/reference/workloads/Debug.cpp
@@ -106,6 +106,12 @@ template void Debug<uint8_t>(const TensorInfo& inputInfo,
                              const std::string& layerName,
                              unsigned int slotIndex);
 
+template void Debug<int8_t>(const TensorInfo& inputInfo,
+                             const int8_t* inputData,
+                             LayerGuid guid,
+                             const std::string& layerName,
+                             unsigned int slotIndex);
+
 template void Debug<int16_t>(const TensorInfo& inputInfo,
                              const int16_t* inputData,
                              LayerGuid guid,
diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp
index 59b836da09..2f0b4276c0 100644
--- a/src/backends/reference/workloads/RefDebugWorkload.cpp
+++ b/src/backends/reference/workloads/RefDebugWorkload.cpp
@@ -48,6 +48,7 @@ template class RefDebugWorkload<DataType::Float16>;
 template class RefDebugWorkload<DataType::Float32>;
 template class RefDebugWorkload<DataType::QAsymmU8>;
 template class RefDebugWorkload<DataType::QSymmS16>;
+template class RefDebugWorkload<DataType::QSymmS8>;
 template class RefDebugWorkload<DataType::Signed32>;
 
 } // namespace armnn
diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp
index 58e4464b00..a15a863892 100644
--- a/src/backends/reference/workloads/RefDebugWorkload.hpp
+++ b/src/backends/reference/workloads/RefDebugWorkload.hpp
@@ -37,10 +37,11 @@ private:
     DebugCallbackFunction m_Callback;
 };
 
-using RefDebugFloat16Workload = RefDebugWorkload<DataType::Float16>;
-using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>;
-using RefDebugQAsymm8Workload = RefDebugWorkload<DataType::QAsymmU8>;
-using RefDebugQSymm16Workload = RefDebugWorkload<DataType::QSymmS16>;
+using RefDebugFloat16Workload  = RefDebugWorkload<DataType::Float16>;
+using RefDebugFloat32Workload  = RefDebugWorkload<DataType::Float32>;
+using RefDebugQAsymm8Workload  = RefDebugWorkload<DataType::QAsymmU8>;
+using RefDebugQSymm16Workload  = RefDebugWorkload<DataType::QSymmS16>;
+using RefDebugQSymm8Workload   = RefDebugWorkload<DataType::QSymmS8>;
 using RefDebugSigned32Workload = RefDebugWorkload<DataType::Signed32>;
 
 } // namespace armnn
diff --git a/src/backends/reference/workloads/RefQuantizeWorkload.cpp b/src/backends/reference/workloads/RefQuantizeWorkload.cpp
index 31534abe3e..ab2ee7fc4e 100644
--- a/src/backends/reference/workloads/RefQuantizeWorkload.cpp
+++ b/src/backends/reference/workloads/RefQuantizeWorkload.cpp
@@ -50,7 +50,7 @@ void RefQuantizeWorkload::Execute() const
         }
         case DataType::QSymmS8:
         {
-            QuantizeImpl<int8_t>(input, output, m_NumElements, m_Scale, m_Offset);
+            QuantizeImpl<int8_t>(input, output, m_NumElements, m_Scale, 0);
             break;
         }
         case DataType::QSymmS16:
-- 
cgit v1.2.1