From 9d9dd223ba9fbc509ea8ff1c211d3c63943a5989 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Fri, 15 Jul 2022 10:22:49 +0100
Subject: IVGCVSW-7107 'Error while running Arm NN Sl with -d option'

* Templated the DumpTensor() function based on tensor type

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I08775e480f89010be61daf0a09a2ab0274e05978
---
 shim/sl/canonical/CanonicalUtils.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'shim/sl/canonical/CanonicalUtils.hpp')

diff --git a/shim/sl/canonical/CanonicalUtils.hpp b/shim/sl/canonical/CanonicalUtils.hpp
index a509684153..b94fd5e126 100644
--- a/shim/sl/canonical/CanonicalUtils.hpp
+++ b/shim/sl/canonical/CanonicalUtils.hpp
@@ -55,10 +55,11 @@ bool isQuantizedOperand(const OperandType& operandType);
 
 std::string GetModelSummary(const Model& model);
 
+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,
-- 
cgit v1.2.1