Use the default rewriter config instead of a custom one
authorBenoit Steiner <bsteiner@google.com>
Mon, 30 Apr 2018 19:13:00 +0000 (12:13 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 30 Apr 2018 19:15:44 +0000 (12:15 -0700)
PiperOrigin-RevId: 194824761

tensorflow/python/grappler/graph_placer.py

index 1cd51df..654013b 100644 (file)
@@ -55,11 +55,6 @@ def PlaceGraph(metagraph,
 
   # Optimize the metagraph to speedup the placement
   rewriter_config = rewriter_config_pb2.RewriterConfig()
-  rewriter_config.optimizers.append("pruning")
-  rewriter_config.optimizers.append("constfold")
-  rewriter_config.optimizers.append("arithmetic")
-  rewriter_config.optimizers.append("dependency")
-  rewriter_config.optimizers.append("pruning")
   optimized_graph = tf_optimizer.OptimizeGraph(
       rewriter_config, metagraph, verbose=verbose, cluster=cluster)
   optimized_metagraph = meta_graph_pb2.MetaGraphDef()