message: Enable shape inference on import for v2 path of MLIR TPU Bridge.
  Certain ops rely on shape inference currently, such as ops that have shapes as derived attributes (e.g. IteratorGetNext). This will allow for more static shapes to be passed to TPUCompile.
diff: @@ -130,12 +130,6 @@ Status MlirFunctionOptimizationPass::Run(
  import_config.graph_as_function = true;
  import_config.control_outputs = *control_ret_node_names;
  import_config.upgrade_legacy = true;
  - // Disable shape inference during import as some TensorFlow op fails during
  - // shape inference with dynamic shaped operands. This in turn causes the
  - // import to fail. Shape inference during import is going to be removed and
  - // the shape inference pass is run early in the pass pipeline, shape inference
  - // during import is not necessary.
  - import_config.enable_shape_inference = false;
  TF_ASSIGN_OR_RETURN(auto module_ref,
  ConvertGraphToMlir(**graph, debug_info, *flib_def,
  import_config, &context));
  