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

1.3/HalPolicy.cpp

diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp
index 1c4a1e3..79df1c7 100644
--- a/1.3/HalPolicy.cpp
+++ b/1.3/HalPolicy.cpp
@@ -111,6 +111,8 @@
             return ConvertQuantizedLstm(operation, model, data);
         case V1_3::OperationType::QUANTIZED_16BIT_LSTM:
             return ConvertQuantized16BitLstm(operation, model, data);
+        case V1_3::OperationType::RANK:
+            return ConvertRank(operation, model, data);
         case V1_3::OperationType::RELU:
             return ConvertReLu(operation, model, data);
         case V1_3::OperationType::RELU1:
@@ -394,6 +396,12 @@
     return ::ConvertQuantized16BitLstm<hal_1_3::HalPolicy>(operation, model, data);
 }
 
+bool HalPolicy::ConvertRank(const Operation& operation, const Model& model, ConversionData& data)
+{
+    ALOGV("hal_1_3::HalPolicy::ConvertRank()");
+    return ::ConvertRank<hal_1_3::HalPolicy>(operation, model, data);
+}
+
 bool HalPolicy::ConvertReLu(const Operation& operation, const Model& model, ConversionData& data)
 {
     ALOGV("hal_1_3::HalPolicy::ConvertReLu()");