commit 4818d465973d72979a7f6e783e1b55b320781710
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Apr 17 11:22:38 2019 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Apr 17 11:55:30 2019 +0100
tree 7c6cce52a728b6fe3ea2733b808555ff27edbab2
parent 0790dcea1056298d63f97dec904c8ade5d21f439

IVGCVSW-2849 Add TfLite Parser support for Rank-0 operands and unit tests

Change-Id: I6dab12aed395a30466d66421c6e5a12659fedac8
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index 86688ad..1ee4950 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -365,9 +365,15 @@
         }
     }
 
+    std::vector<unsigned int> safeShape = shapes;
+    if (safeShape.size() == 0)
+    {
+        safeShape.push_back(1);
+    }
+
     // two statements (on purpose) for easier debugging:
-    armnn::TensorInfo result(static_cast<unsigned int>(shapes.size()),
-                             shapes.data(),
+    armnn::TensorInfo result(static_cast<unsigned int>(safeShape.size()),
+                             safeShape.data(),
                              type,
                              quantizationScale,
                              quantizationOffset);