aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2021-11-09 15:43:37 +0000
committerDavid Monahan <David.Monahan@arm.com>2021-11-09 18:26:07 +0000
commitbc8cacae736c569fd3caf343e7ad7a9e5534bf27 (patch)
treedda366cdd464cc35bb3048721597be571365285a /Utils.hpp
parentee6818be7815e10be4535645f0472ae5ad116309 (diff)
downloadandroid-nn-driver-branches/android-nn-driver_21_11.tar.gz
Fixed Driver Crash in DumpTensorv21.11branches/android-nn-driver_21_11
* 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
Diffstat (limited to 'Utils.hpp')
-rw-r--r--Utils.hpp3
1 files changed, 2 insertions, 1 deletions
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,