1 files changed, 6 insertions, 0 deletions
diff --git a/Utils.cpp b/Utils.cpp
index c01604c3..6ee025e6 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -6,6 +6,7 @@
 #define LOG_TAG "ArmnnDriver"
 
 #include "Utils.hpp"
+#include "Half.hpp"
 
 #include <armnnUtils/Permute.hpp>
 
@@ -241,6 +242,11 @@ void DumpTensor(const std::string& dumpDir,
             dumpElementFunction = &DumpTensorElement<int32_t>;
             break;
         }
+        case armnn::DataType::Float16:
+        {
+            dumpElementFunction = &DumpTensorElement<armnn::Half>;
+            break;
+        }
         default:
         {
             dumpElementFunction = nullptr;