message: Add GuaranteeAllFuncsOneUse pass to bridge pass pipeline.
  This enables shape inference where functions are used multiple times.
diff: @@ -304,6 +304,8 @@ Status ConvertMLIRToXlaComputation(
  tf2xla.addPass(mlir::TFDevice::CreateDecomposeResourceOpsPass());
  tf2xla.addPass(mlir::TF::CreatePromoteResourcesToArgsPass());
  tf2xla.addPass(mlir::createSymbolDCEPass());
  + // Guarantee all functions have one use, which enables shape inference.
  + tf2xla.addPass(mlir::TF::CreateGuaranteeAllFuncsOneUsePass());
  tf2xla.addPass(mlir::TF::CreateTFShapeInferencePass());
  // LegalizeTFControlFlow encapsulates arguments for control flow operations
  // with a tuple argument which break the assumption of resource lifting
  