[tfltool] Enable select operator for Pack op (#3817)

This patch will enable select operator tflitetool for pack op

Signed-off-by: shubham <shub98.gupta@samsung.com>

diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
index b615695..c5d311d 100644 (file)
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -725,7 +725,19 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
     # PowOptions: not supported
     # ArgMinOptions: not supported
     # FakeQuantOptions: not supported
-    # PackOptions: not supported
+
+    # PackOptions:
+    import tflite.PackOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().PackOptions:
+
+        pack_option = tflite.PackOptions.PackOptions()
+        pack_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.PackOptions.PackOptionsStart(new_builder)
+        tflite.PackOptions.PackOptionsAddValuesCount(new_builder,
+                                                     pack_option.ValuesCount())
+        tflite.PackOptions.PackOptionsAddAxis(new_builder, pack_option.Axis())
+        return tflite.PackOptions.PackOptionsEnd(new_builder)
 
     # LogicalOrOptions:
     import tflite.LogicalOrOptions