[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 (executable)
--- 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