From caf2bdd3dd64ece480add64a21ac4653b6e0a26e Mon Sep 17 00:00:00 2001
From: Tee Jung <tee.ty.jung@openedges.com>
Date: Wed, 13 Nov 2019 07:23:14 +0000
Subject: 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
---
 src/armnnOnnxParser/OnnxParser.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

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);
 }
 
-- 
cgit v1.2.1