[tfltool] Enable select operator for transpose op (#3811)

This patch will enable transpose op in tflitetool in select_operator.py

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 0100b64..b615695 100644 (file)
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -501,7 +501,15 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         tflite.SpaceToBatchNDOptions.SpaceToBatchNDOptionsStart(new_builder)
         return tflite.SpaceToBatchNDOptions.SpaceToBatchNDOptionsEnd(new_builder)
 
-    # TransposeOptions: not supported
+    # TransposeOptions:
+    import tflite.TransposeOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().TransposeOptions:
+
+        transpose_option = tflite.TransposeOptions.TransposeOptions()
+        transpose_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.TransposeOptions.TransposeOptionsStart(new_builder)
+        return tflite.TransposeOptions.TransposeOptionsEnd(new_builder)
 
     # ReducerOptions
     import tflite.ReducerOptions