aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2021-11-09 15:43:37 +0000
committerMike Kelly <mike.kelly@arm.com>2021-11-09 15:43:37 +0000
commit37c3e508dcd767e8823c913f3f487d6b0bd03fb2 (patch)
treedda366cdd464cc35bb3048721597be571365285a /Utils.hpp
parentee6818be7815e10be4535645f0472ae5ad116309 (diff)
downloadandroid-nn-driver-37c3e508dcd767e8823c913f3f487d6b0bd03fb2.tar.gz
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
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,