Enable select operator for reduce operators: reduce_max, reduce_min, reduce_any, mean, sum,
Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
return tflite.SpaceToBatchNDOptions.SpaceToBatchNDOptionsEnd(new_builder)
# TransposeOptions: not supported
- # ReducerOptions: not supported
+
+ # ReducerOptions
+ import tflite.ReducerOptions
+ if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ReducerOptions:
+
+ reducer_option = tflite.ReducerOptions.ReducerOptions()
+ reducer_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+ tflite.ReducerOptions.ReducerOptionsStart(new_builder)
+ tflite.ReducerOptions.ReducerOptionsAddKeepDims(new_builder,
+ reducer_option.KeepDims())
+ return tflite.ReducerOptions.ReducerOptionsEnd(new_builder)
# SubOptions
import tflite.SubOptions