diff options
| -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); } |