diff options
Diffstat (limited to 'tools/tflitefile_tool')
-rwxr-xr-x | tools/tflitefile_tool/select_operator.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index 42854c2c0..48d814563 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -643,6 +643,18 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t # ArgMaxOptions: not supported # LessOptions: not supported + # ArgMaxOptions + import tflite.ArgMaxOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ArgMaxOptions: + + arg_max_option = tflite.ArgMaxOptions.ArgMaxOptions() + arg_max_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + tflite.ArgMaxOptions.ArgMaxOptionsStart(new_builder) + tflite.ArgMaxOptions.ArgMaxOptionsAddOutputType(new_builder, + arg_max_option.OutputType()) + return tflite.ArgMaxOptions.ArgMaxOptionsEnd(new_builder) + # NegOptions import tflite.NegOptions if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().NegOptions: |