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
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,