diff options
Diffstat (limited to 'src/backends/reference/workloads')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 5 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.cpp | 1 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.hpp | 3 |
3 files changed, 8 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index 594f428908..d1c9fdd8b8 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -97,4 +97,9 @@ template void Debug<uint8_t>(const TensorInfo& inputInfo, const std::string& layerName, unsigned int slotIndex); +template void Debug<int16_t>(const TensorInfo& inputInfo, + const int16_t* inputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); } // namespace armnn diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp index be2d82f6dc..325817b19f 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.cpp +++ b/src/backends/reference/workloads/RefDebugWorkload.cpp @@ -46,5 +46,6 @@ void RefDebugWorkload<DataType>::RegisterDebugCallback(const DebugCallbackFuncti template class RefDebugWorkload<DataType::Float32>; template class RefDebugWorkload<DataType::QuantisedAsymm8>; +template class RefDebugWorkload<DataType::QuantisedSymm16>; } // namespace armnn diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp index 2985699f7b..6a1fceba0a 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.hpp +++ b/src/backends/reference/workloads/RefDebugWorkload.hpp @@ -38,6 +38,7 @@ private: }; using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>; -using RefDebugUint8Workload = RefDebugWorkload<DataType::QuantisedAsymm8>; +using RefDebugQAsymm8Workload = RefDebugWorkload<DataType::QuantisedAsymm8>; +using RefDebugQSymm16Workload = RefDebugWorkload<DataType::QuantisedSymm16>; } // namespace armnn |