diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-11-12 16:15:11 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-15 11:57:50 +0000 |
commit | db1a2834b4e9d74ed538943634212eccbd4a789b (patch) | |
tree | 10162ed219468bc043ed66e40ea657fb867b0ee1 /src/backends/reference/workloads/Debug.cpp | |
parent | 60538ada2b90704abcf6473144639103d80287a5 (diff) | |
download | armnn-db1a2834b4e9d74ed538943634212eccbd4a789b.tar.gz |
Add FP16 support to DebugWorkload
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ia879f2d84a1b977474ee0dafa976f2aab32bd3ae
Diffstat (limited to 'src/backends/reference/workloads/Debug.cpp')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 10 |
1 files changed, 10 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 |