message: Disable lower using switch and merge in grappler optimization
diff: @@ -1031,10 +1031,16 @@ class TFLiteConverter(TFLiteConverterBase):
  (self.inference_type == constants.INT8 and
  (post_training_optimize or weight_only_quantize))):
  try:
  + # TODO(b/150163103): Merge `disabling lower using switch merge' calls.
  + # Grappler will also try to lower while loop into switch merge
  + # representation which is undesired for Ophints, so we simply remove
  + # those attributes to prevent Grappler from doing so.
  + graph_def = _convert_to_constants.disable_lower_using_switch_merge(
  + optimized_graph)
  # Run function inlining optimization to ensure any models generated
  # through the from_frozen_graph path have been inlined.
  optimized_graph = _run_graph_optimizations(
  - self._graph_def,
  + graph_def,
  self._input_tensors,
  self._output_tensors,
  config=self._grappler_config(["function"]))
  