From 120cc36f0e19c7d899ad99ea8b41d808f984ac1e Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Mon, 25 Mar 2019 11:18:05 -0700
Subject: [PATCH] Second invocation of the model pruner after all other
 optimizers removes too many nodes in some cases in TF 2.0 mode.

PiperOrigin-RevId: 240180399
---
 tensorflow/core/grappler/optimizers/meta_optimizer.cc | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc
index b4a17e27317..9c13562f3d2 100644
--- a/tensorflow/core/grappler/optimizers/meta_optimizer.cc
+++ b/tensorflow/core/grappler/optimizers/meta_optimizer.cc
@@ -447,10 +447,6 @@ Status MetaOptimizer::OptimizeGraph(Cluster* cluster, const GrapplerItem& item,
   optimization_results_.push_back(optimization_result);
 
   if (is_optimized) {
-    // Run the model pruner again to clean things up.
-    if (!cfg_.disable_model_pruning() && model_pruner != nullptr) {
-      RUN_OPTIMIZER_OR_RETURN_IF_ERROR(model_pruner);
-    }
     TF_RETURN_IF_ERROR(TopologicalSort(optimized_graph));
     ReassignColocation(optimized_graph);
     // Make sure that the optimizers preserved the graph version.