From 3262f347aea3681bf0bde0f8cc4c121eeef85f75 Mon Sep 17 00:00:00 2001
From: Bixia Zheng <bixia@google.com>
Date: Tue, 4 Feb 2020 15:09:38 -0800
Subject: [PATCH] [TF:MLIR] Add the symbol DCE pass to remove dead functions
 after inlining.

PiperOrigin-RevId: 293238742
Change-Id: I5130d30acbfe81423572ba8bc46c0151a22615f4
---
 tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc b/tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
index a52b30e2fd2..c2f17d6e38c 100644
--- a/tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
+++ b/tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
@@ -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());