commit 9d9dd223ba9fbc509ea8ff1c211d3c63943a5989
author Sadik Armagan <sadik.armagan@arm.com> Fri Jul 15 10:22:49 2022 +0100
committer Nikhil Raj <nikhil.raj@arm.com> Wed Jul 27 15:56:19 2022 +0100
tree 24a3b084853ed954dc4098782a2ee6a8bf4fc95c
parent 851df867f263e23daa5139a5e672b03d4da08266

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

diff --git a/shim/sl/canonical/CanonicalUtils.hpp b/shim/sl/canonical/CanonicalUtils.hpp
index a509684..b94fd5e 100644
--- a/shim/sl/canonical/CanonicalUtils.hpp
+++ b/shim/sl/canonical/CanonicalUtils.hpp
@@ -55,10 +55,11 @@
 
 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,