diff options
author | Tee Jung <tee.ty.jung@openedges.com> | 2019-11-13 07:23:14 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-15 20:55:02 +0000 |
commit | caf2bdd3dd64ece480add64a21ac4653b6e0a26e (patch) | |
tree | 77c1d3213aa66c19dacbdf640d33e893f6a9cab7 | |
parent | 17d8d85ece077c1272839edea32c55af553ced59 (diff) | |
download | armnn-caf2bdd3dd64ece480add64a21ac4653b6e0a26e.tar.gz |
Fix possible crash in case of zero dimension tensor in the ONNX
parser
Signed-off-by: Jung Tae-young <tee.ty.jung@openedges.com>
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Change-Id: I396792d4d59172cccb50d77de7e6b74977b289ed
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.cpp b/src/armnnOnnxParser/OnnxParser.cpp index 0d0cc253d2..42bc427638 100644 --- a/src/armnnOnnxParser/OnnxParser.cpp +++ b/src/armnnOnnxParser/OnnxParser.cpp @@ -215,6 +215,13 @@ armnn::TensorInfo ToTensorInfo(const std::string& name, std::vector<unsigned int CHECK_LOCATION().AsString() )); } } + + // To avoid crashes by trivial tensors + if (shape.empty()) + { + return TensorInfo(TensorShape(), type); + } + return TensorInfo(TensorShape(static_cast<unsigned int>(shape.size()), shape.data()), type); } |