commit fc884b4141a28fbd3c62f665341ec88158fcd332
author Finn Williams <Finn.Williams@Arm.com> Thu Jun 11 17:35:44 2020 +0100
committer Finn Williams <Finn.Williams@Arm.com> Mon Jul 06 19:07:52 2020 +0100
tree df18555baced3f4d917a78d24d87c5e2d1778f83
parent 00b586b1f3106a4e6970ca7feacb1cc1892d107e

Add support for Rank and scalar tensors

!armnn:3330

Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: Icc429d9fabb570193d12bffef0e00dda7b51032f

Utils.hpp

diff --git a/Utils.hpp b/Utils.hpp
index b61ddb2..d58d273 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -150,18 +150,24 @@
 {
     V1_2::OutputShape shape;
 
-    android::hardware::hidl_vec<uint32_t> dimensions;
-
     armnn::TensorShape tensorShape = info.GetShape();
-    const unsigned int numDims = tensorShape.GetNumDimensions();
-    dimensions.resize(numDims);
-
-    for (unsigned int outputIdx = 0u; outputIdx < numDims; ++outputIdx)
+    // Android will expect scalars as a zero dimensional tensor
+    if(tensorShape.GetDimensionality() == armnn::Dimensionality::Scalar)
     {
-        dimensions[outputIdx] = tensorShape[outputIdx];
+         shape.dimensions = android::hardware::hidl_vec<uint32_t>{};
+    }
+    else
+    {
+        android::hardware::hidl_vec<uint32_t> dimensions;
+        const unsigned int numDims = tensorShape.GetNumDimensions();
+        dimensions.resize(numDims);
+        for (unsigned int outputIdx = 0u; outputIdx < numDims; ++outputIdx)
+        {
+            dimensions[outputIdx] = tensorShape[outputIdx];
+        }
+        shape.dimensions = dimensions;
     }
 
-    shape.dimensions = dimensions;
     shape.isSufficient = true;
 
     return shape;