diff options
Diffstat (limited to 'src/backends/reference/workloads')
4 files changed, 13 insertions, 5 deletions
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: |