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

[TF:MLIR] Add the symbol DCE pass to remove dead functions after inlining.

Browse Source 
PiperOrigin-RevId: 293238742
Change-Id: I5130d30acbfe81423572ba8bc46c0151a22615f4
This commit is contained in:
Bixia Zheng 2020-02-04 15:09:38 -08:00 committed by TensorFlower Gardener
parent 5f56c0fbd8
commit 3262f347ae
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,6 +63,7 @@ void CreateTFStandardPipeline(OpPassManager &pm,
if (options.enable_inliner) {
pm.addPass(createInlinerPass());
}
pm.addPass(createSymbolDCEPass());
pm.addPass(CreateTFShapeInferencePass());
pm.addNestedPass<FuncOp>(CreateTFOptimizePass());
pm.addNestedPass<FuncOp>(createCSEPass());