diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-07-15 10:22:49 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-07-27 15:56:19 +0100 |
commit | 9d9dd223ba9fbc509ea8ff1c211d3c63943a5989 (patch) | |
tree | 24a3b084853ed954dc4098782a2ee6a8bf4fc95c /shim/sl/canonical/CanonicalUtils.hpp | |
parent | 851df867f263e23daa5139a5e672b03d4da08266 (diff) | |
download | armnn-9d9dd223ba9fbc509ea8ff1c211d3c63943a5989.tar.gz |
IVGCVSW-7107 'Error while running Arm NN Sl with -d option'
* Templated the DumpTensor() function based on tensor type
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I08775e480f89010be61daf0a09a2ab0274e05978
Diffstat (limited to 'shim/sl/canonical/CanonicalUtils.hpp')
-rw-r--r-- | shim/sl/canonical/CanonicalUtils.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/shim/sl/canonical/CanonicalUtils.hpp b/shim/sl/canonical/CanonicalUtils.hpp index a509684153..b94fd5e126 100644 --- a/shim/sl/canonical/CanonicalUtils.hpp +++ b/shim/sl/canonical/CanonicalUtils.hpp @@ -55,10 +55,11 @@ bool isQuantizedOperand(const OperandType& operandType); std::string GetModelSummary(const Model& model); +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, |