Enable TPUExtractOutsideCompilationPass in MLIR TPU Bridge.

PiperOrigin-RevId: 327867890 Change-Id: I1e4543fa2e3aa2045bfde72c3f195c0db4829672

Enable TPUExtractOutsideCompilationPass in MLIR TPU Bridge.
PiperOrigin-RevId: 327867890 Change-Id: I1e4543fa2e3aa2045bfde72c3f195c0db4829672
c99c08b9 · Ken Franko · TensorFlower Gardener · 2f006929 · c99c08b9
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc +1 -0

未找到文件。
--- a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
+++ b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
@@ -103,6 +103,7 @@ void CreateTPUBridgePipeline(OpPassManager &pm) {
  pm.addPass(mlir::createInlinerPass());
  pm.addPass(TFDevice::CreateMarkOpsForOutsideCompilationPass());
  pm.addPass(CreateTPUExtractHeadTailOutsideCompilationPass());
+  pm.addPass(CreateTPUExtractOutsideCompilationPass());
  pm.addPass(TF::CreateTFRegionControlFlowToFunctional());

  pm.addNestedPass<FuncOp>(tf_executor::CreateTFExecutorConstantSinkingPass());