diff options
author | Mike Kelly <mike.kelly@arm.com> | 2021-11-09 15:43:37 +0000 |
---|---|---|
committer | Mike Kelly <mike.kelly@arm.com> | 2021-11-09 15:43:37 +0000 |
commit | 37c3e508dcd767e8823c913f3f487d6b0bd03fb2 (patch) | |
tree | dda366cdd464cc35bb3048721597be571365285a /Utils.hpp | |
parent | ee6818be7815e10be4535645f0472ae5ad116309 (diff) | |
download | android-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.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, |