diff options
Diffstat (limited to 'src/backends/reference/workloads/Debug.cpp')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index 49e9e02ffb..aadbc7613b 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -5,6 +5,7 @@ #include "Debug.hpp" +#include <BFloat16.hpp> #include <Half.hpp> #include <boost/numeric/conversion/cast.hpp> @@ -88,6 +89,12 @@ void Debug(const TensorInfo& inputInfo, std::cout << " }" << std::endl; } +template void Debug<BFloat16>(const TensorInfo& inputInfo, + const BFloat16* inputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); + template void Debug<Half>(const TensorInfo& inputInfo, const Half* inputData, LayerGuid guid, |