From: Prasanna R/System SW /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Date: Tue, 16 Oct 2018 01:25:41 +0000 (+0530)
Subject: [tfltool] OP selection: Enable for EqualOptions (#3112)
X-Git-Tag: 0.3~637
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e8cc13e9335bb823e7b02f9f2cd588fdaaf49ed0;p=platform%2Fcore%2Fml%2Fnnfw.git

[tfltool] OP selection: Enable for EqualOptions (#3112)

This patch will enable EqualOptions in tfltool select_operator.
Related issue: #2610 #2822

Signed-off-by: prasannar <prasanna.r@samsung.com>
---

diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
index 54586ae..ded6761 100755
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -513,6 +513,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         tflite.NegOptions.NegOptionsStart(new_builder)
         return tflite.NegOptions.NegOptionsEnd(new_builder)
 
+    # EqualOptions
+    import tflite.EqualOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().EqualOptions:
+
+        equal_option = tflite.EqualOptions.EqualOptions()
+        equal_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.EqualOptions.EqualOptionsStart(new_builder)
+        return tflite.EqualOptions.EqualOptionsEnd(new_builder)
+
     # PadV2Options: not supported
     # GreaterOptions: not supported
     # GreaterEqualOptions: not supported
@@ -523,7 +533,6 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
     # SparseToDenseOptions: not supported
     # TileOptions: not supported
     # ExpandDimsOptions: not supported
-    # EqualOptions: not supported
     # NotEqualOptions: not supported
     # ShapeOptions: not supported
     # PowOptions: not supported