From 37c3e508dcd767e8823c913f3f487d6b0bd03fb2 Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Tue, 9 Nov 2021 15:43:37 +0000
Subject: Fixed Driver Crash in DumpTensor

 * When handling Tensors DumpTensor was automatically trying to turn
   them into ConstTensors but then threw an exceptions when IsConstant
   returned false

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I8681bb3dd41cfe19c60fbd1cc9394c8b6cca551e
---
 Utils.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'Utils.hpp')

diff --git a/Utils.hpp b/Utils.hpp
index 9bd28ba6..6e733a26 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -126,10 +126,11 @@ std::string GetModelSummary(const HalModel& model)
     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,
-- 
cgit v1.2.1