Add support for Rank and scalar tensors
!armnn:3330
Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: Icc429d9fabb570193d12bffef0e00dda7b51032f
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()");
diff --git a/1.3/HalPolicy.hpp b/1.3/HalPolicy.hpp
index 6df2ce2..0eb5f4d 100644
--- a/1.3/HalPolicy.hpp
+++ b/1.3/HalPolicy.hpp
@@ -123,6 +123,8 @@
static bool ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data);
+ static bool ConvertRank(const Operation& operation, const Model& model, ConversionData& data);
+
static bool ConvertReLu(const Operation& operation, const Model& model, ConversionData& data);
static bool ConvertReLu1(const Operation& operation, const Model& model, ConversionData& data);