diff options
author | Mike Kelly <mike.kelly@arm.com> | 2021-11-09 15:43:37 +0000 |
---|---|---|
committer | David Monahan <David.Monahan@arm.com> | 2021-11-09 18:26:07 +0000 |
commit | bc8cacae736c569fd3caf343e7ad7a9e5534bf27 (patch) | |
tree | dda366cdd464cc35bb3048721597be571365285a /Utils.hpp | |
parent | ee6818be7815e10be4535645f0472ae5ad116309 (diff) | |
download | android-nn-driver-bc8cacae736c569fd3caf343e7ad7a9e5534bf27.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.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -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, |