diff options
Diffstat (limited to 'src/backends/reference/workloads')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 10 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.cpp | 1 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.hpp | 1 |
3 files changed, 12 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index 09a0dfc03b..b7d0911ef3 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -2,8 +2,11 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // + #include "Debug.hpp" +#include <Half.hpp> + #include <boost/numeric/conversion/cast.hpp> #include <algorithm> @@ -85,6 +88,12 @@ void Debug(const TensorInfo& inputInfo, std::cout << " }" << std::endl; } +template void Debug<Half>(const TensorInfo& inputInfo, + const Half* inputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); + template void Debug<float>(const TensorInfo& inputInfo, const float* inputData, LayerGuid guid, @@ -102,4 +111,5 @@ template void Debug<int16_t>(const TensorInfo& inputInfo, 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 325817b19f..2a3883f8f7 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.cpp +++ b/src/backends/reference/workloads/RefDebugWorkload.cpp @@ -44,6 +44,7 @@ void RefDebugWorkload<DataType>::RegisterDebugCallback(const DebugCallbackFuncti m_Callback = func; } +template class RefDebugWorkload<DataType::Float16>; template class RefDebugWorkload<DataType::Float32>; template class RefDebugWorkload<DataType::QuantisedAsymm8>; template class RefDebugWorkload<DataType::QuantisedSymm16>; diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp index 6a1fceba0a..0964515b2c 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.hpp +++ b/src/backends/reference/workloads/RefDebugWorkload.hpp @@ -37,6 +37,7 @@ private: DebugCallbackFunction m_Callback; }; +using RefDebugFloat16Workload = RefDebugWorkload<DataType::Float16>; using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>; using RefDebugQAsymm8Workload = RefDebugWorkload<DataType::QuantisedAsymm8>; using RefDebugQSymm16Workload = RefDebugWorkload<DataType::QuantisedSymm16>; |