diff --git a/Utils.hpp b/Utils.hpp
index 9bd28ba..6e733a2 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -126,10 +126,11 @@
     return result.str();
 }
 
+template <typename TensorType>
 void DumpTensor(const std::string& dumpDir,
                 const std::string& requestName,
                 const std::string& tensorName,
-                const armnn::ConstTensor& tensor);
+                const TensorType& tensor);
 
 void DumpJsonProfilingIfRequired(bool gpuProfilingEnabled,
                                  const std::string& dumpDir,