From: Shubham Gupta/SNAP /SRI-Bangalore/Engineer/삼성전자 Date: Thu, 6 Dec 2018 00:57:23 +0000 (+0530) Subject: [tfltool] Enable select operator for transpose op (#3811) X-Git-Tag: 0.3~190 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b207a7cc8eda9850efa9ce7d719900a2faf9a7d4;p=platform%2Fcore%2Fml%2Fnnfw.git [tfltool] Enable select operator for transpose op (#3811) This patch will enable transpose op in tflitetool in select_operator.py Signed-off-by: shubham --- diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index 0100b64..b615695 100644 --- 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