experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable MarkOpsForOutsideCompilation pass in TPU bridge.

Browse Source 
PiperOrigin-RevId: 327540960
Change-Id: I821d6f1a233219cc8da6d76c7d6934eb1bec01cd
This commit is contained in:
Ken Franko 2020-08-19 17:35:06 -07:00 committed by TensorFlower Gardener
parent 8257866ced
commit 78635aa358
1 changed files with 1 additions and 0 deletions

1
tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
View File

@ -101,6 +101,7 @@ void CreateTPUBridgePipeline(OpPassManager &pm) {
pm.addPass(TFDevice::CreateResourceOpLiftingPass());
pm.addPass(TF::CreateTFFunctionalControlFlowToRegions());
pm.addPass(mlir::createInlinerPass());
pm.addPass(TFDevice::CreateMarkOpsForOutsideCompilationPass());
pm.addPass(CreateTPUExtractHeadTailOutsideCompilationPass());
pm.addPass(TF::CreateTFRegionControlFlowToFunctional());