[tfltool] Correction in select operator for ArgMax op (#3854)
authorDevansh Bansal/SNAP /SRI-Bangalore/Engineer/삼성전자 <b.devansh@samsung.com>
Tue, 4 Dec 2018 11:01:36 +0000 (16:31 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 4 Dec 2018 11:01:36 +0000 (20:01 +0900)
This patch removes "not supported" for ArgMax in tflitetool in select_operator.py

Signed-off-by: b.devansh <b.devansh@samsung.com>
tools/tflitefile_tool/select_operator.py

index ca6b1f5..3469041 100755 (executable)
@@ -640,8 +640,6 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         return tflite.DequantizeOptions.DequantizeOptionsEnd(new_builder)
 
     # MaximumMinimumOptions: not supported
-    # ArgMaxOptions: not supported
-    # LessOptions: not supported
 
     # ArgMaxOptions
     import tflite.ArgMaxOptions
@@ -655,6 +653,8 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
                                                         arg_max_option.OutputType())
         return tflite.ArgMaxOptions.ArgMaxOptionsEnd(new_builder)
 
+    # LessOptions: not supported
+
     # NegOptions
     import tflite.NegOptions
     if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().NegOptions: