message: Add Outside Compilation Head/Tail Extraction related passes to TPU MLIR bridge pipeline.
diff: @@ -87,7 +87,8 @@ void CreateTPUBridgePipeline(OpPassManager &pm) {
  // because DecomposeResourceOpsPass uses pattern rewriter which hoists
  // changed constants out of tf_device.Launch.
  func_pm.addPass(TFDevice::CreateDecomposeResourceOpsPass());
  -
  + pm.addNestedPass<FuncOp>(CreateTPUHostComputationExpansionPass());
  + pm.addPass(CreateTPUExtractHeadTailOutsideCompilationPass());
  // Run another shape inference pass because resource decomposition might have
  // created new partial types.
  pm.addPass(TF::CreateTFShapeInferencePass());
  