[tflitefile_tool] select_operator supports Select op (#3915)
authorSanggyu Lee <takepencil@naver.com>
Mon, 24 Aug 2020 02:44:15 +0000 (11:44 +0900)
committerGitHub <noreply@github.com>
Mon, 24 Aug 2020 02:44:15 +0000 (11:44 +0900)
select_operator will support Select op.

ONE-DCO-1.0-Signed-off-by: Sanggyu Lee <sg5.lee@samsung.com>

tools/tflitefile_tool/select_operator.py

index 628e477..dcafd75 100755 (executable)
@@ -723,7 +723,17 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
     # GreaterOptions: not supported
     # GreaterEqualOptions: not supported
     # LessEqualOptions: not supported
-    # SelectOptions: not supported
+
+    # SelectOptions
+    import tflite.SelectOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().SelectOptions:
+
+        select_option = tflite.SelectOptions.SelectOptions()
+        select_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.SelectOptions.SelectOptionsStart(new_builder)
+        return tflite.SelectOptions.SelectOptionsEnd(new_builder)
+
     # SliceOptions: not supported
 
     # TransposeConvOptions