From 2fc21f7f64ba0538add9b5f384661e8b28dd1acb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89anna=20=C3=93=20Cath=C3=A1in?=
 <eanna.ocathain@arm.com>
Date: Mon, 13 May 2019 11:01:33 +0100
Subject: IVGCVSW-3064 Implementing our own version of the
 compliantWithV1_0(operation)              and convertToV1_0(operation)
 functions
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Change-Id: I2c0927013e16baefcc7290bbcfc8e76c4a645c35
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
---
 1.1/HalPolicy.cpp | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 51 insertions(+), 2 deletions(-)

diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp
index 9a0c1bf2..9acd01c6 100644
--- a/1.1/HalPolicy.cpp
+++ b/1.1/HalPolicy.cpp
@@ -7,6 +7,55 @@
 
 #include "../1.0/HalPolicy.hpp"
 
+namespace
+{
+static std::vector<V1_0::OperationType> opsEquivalentInV10({
+    V1_0::OperationType::ADD,
+    V1_0::OperationType::AVERAGE_POOL_2D,
+    V1_0::OperationType::CONCATENATION,
+    V1_0::OperationType::CONV_2D,
+    V1_0::OperationType::DEPTHWISE_CONV_2D,
+    V1_0::OperationType::FLOOR,
+    V1_0::OperationType::FULLY_CONNECTED,
+    V1_0::OperationType::LOCAL_RESPONSE_NORMALIZATION,
+    V1_0::OperationType::LOGISTIC,
+    V1_0::OperationType::LSTM,
+    V1_0::OperationType::L2_NORMALIZATION,
+    V1_0::OperationType::L2_POOL_2D,
+    V1_0::OperationType::MAX_POOL_2D,
+    V1_0::OperationType::MUL,
+    V1_0::OperationType::RELU,
+    V1_0::OperationType::RELU1,
+    V1_0::OperationType::RELU6,
+    V1_0::OperationType::SOFTMAX,
+    V1_0::OperationType::TANH,
+    V1_0::OperationType::RESHAPE,
+    V1_0::OperationType::RESIZE_BILINEAR,
+});
+
+bool CompliantWithVersion10(const V1_1::Operation & operation)
+{
+    std::vector<V1_0::OperationType>::iterator it;
+    it = std::find(opsEquivalentInV10.begin(), opsEquivalentInV10.end(),
+                   static_cast<V1_0::OperationType>(operation.type));
+
+    if(it != opsEquivalentInV10.end())
+    {
+        return true;
+    }
+    return false;
+}
+
+V1_0::Operation ConvertOperationToVersion10(const V1_1::Operation & operation)
+{
+    V1_0::Operation v10Operation;
+    v10Operation.type = static_cast<V1_0::OperationType>(operation.type);
+    v10Operation.inputs = operation.inputs;
+    v10Operation.outputs = operation.outputs;
+    return v10Operation;
+}
+}
+
 namespace armnn_driver
 {
 namespace hal_1_1
@@ -14,9 +63,9 @@ namespace hal_1_1
 
 bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, ConversionData& data)
 {
-    if (compliantWithV1_0(operation))
+    if (CompliantWithVersion10(operation))
     {
-        hal_1_0::HalPolicy::Operation v10Operation = convertToV1_0(operation);
+        hal_1_0::HalPolicy::Operation v10Operation = ConvertOperationToVersion10(operation);
         hal_1_0::HalPolicy::Model v10Model = convertToV1_0(model);
 
         return hal_1_0::HalPolicy::ConvertOperation(v10Operation, v10Model, data);
-- 
cgit v1.2.1