new_builder, add_option.FusedActivationFunction())
return tflite.AddOptions.AddOptionsEnd(new_builder)
- # L2NormOptions: not supported
+ # L2NormOptions
+ import tflite.L2NormOptions
+ if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().L2NormOptions:
+
+ l2norm_option = tflite.L2NormOptions.L2NormOptions()
+ l2norm_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+ tflite.L2NormOptions.L2NormOptionsStart(new_builder)
+ tflite.L2NormOptions.L2NormOptionsAddFusedActivationFunction(
+ new_builder, L2NormOptions.FusedActivationFunction())
+ return tflite.L2NormOptions.L2NormOptionsEnd(new_builder)
+
# LocalResponseNormalizationOptions: not supported
# LSTMOptions: not supported
tflite.TopKV2Options.TopKV2OptionsStart(new_builder)
return tflite.TopKV2Options.TopKV2OptionsEnd(new_builder)
- # SplitOptions: not supported
+ # SplitOptions
+ import tflite.SplitOptions
+ if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().SplitOptions:
+
+ split_option = tflite.SplitOptions.SplitOptions()
+ split_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+ tflite.SplitOptions.SplitOptionsStart(new_builder)
+ tflite.SplitOptions.SplitOptionsAddNumSplits(new_builder,
+ SplitOptions.NumSplits())
+ return tflite.SplitOptions.SplitOptionsEnd(new_builder)
+
# LogSoftmaxOptions: not supported
# CastOptions: not supported