message: Reorder kernel gen passes.
  This is due to moving patterns between the two  passes to allow proper unranked handling of both binary and special unary functions.
diff: @@ -87,8 +87,8 @@ Status LowerTFtoGPU(mlir::ModuleOp module, bool gpu_binary_only,
  } else {
  pm.addPass(mlir::mhlo::createLegalizeTFPass(
  /*allow_partial_conversion=*/false, /*legalize_chlo=*/false));
  - pm.addPass(mlir::mhlo::createChloLegalizeToHloPass());
  pm.addPass(mlir::createTransformUnrankedHloPass());
  + pm.addPass(mlir::mhlo::createChloLegalizeToHloPass());
  pm.addPass(mlir::kernel_gen::transforms::CreateShapeToDescriptorsPass());
  // Clean up the IR created above. In particular, operations on descriptors
  // are simplified here.
  