Automated g4 rollback of changelist 187563544

PiperOrigin-RevId: 187582263

1 files changed, 1 insertions, 28 deletions

diff --git a/tensorflow/python/grappler/memory_optimizer_test.py b/tensorflow/python/grappler/memory_optimizer_test.py
index 58d3c1e85f..948911f099 100644
--- a/tensorflow/python/grappler/memory_optimizer_test.py
+++ b/tensorflow/python/grappler/memory_optimizer_test.py
@@ -162,34 +162,7 @@ class MemoryOptimizerRecomputeTest(test.TestCase):
             arithmetic_optimization=rewriter_config_pb2.RewriterConfig.OFF,
             memory_optimization=rewriter_config_pb2.RewriterConfig.
             RECOMPUTATION_HEURISTICS,
-            memory_optimizer_target_node_name_regexp='optimizer/gradients/'),
-        original_metagraph)
-    self.assertGreater(
-        len(rewritten_graph_def.node),
-        len(original_metagraph.graph_def.node))
-    self.assertEqual(
-        0,
-        len([node for node in original_metagraph.graph_def.node
-             if 'Recomputed/' in node.name]))
-    self.assertEqual(
-        20,  # Two per layer
-        len([node for node in rewritten_graph_def.node
-             if 'Recomputed/' in node.name]))
-
-  def testRewritingNameScopedGradientNamesRegexp(self):
-    """Tests that rewriting occurs with non-standard gradient names."""
-    (original_metagraph, _, _, _) = self._GetMetaGraph(
-        optimizer_scope_name='foo/bar')
-    rewritten_graph_def = tf_optimizer.OptimizeGraph(
-        rewriter_config_pb2.RewriterConfig(
-            disable_model_pruning=True,
-            constant_folding=rewriter_config_pb2.RewriterConfig.OFF,
-            dependency_optimization=rewriter_config_pb2.RewriterConfig.OFF,
-            layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF,
-            arithmetic_optimization=rewriter_config_pb2.RewriterConfig.OFF,
-            memory_optimization=rewriter_config_pb2.RewriterConfig.
-            RECOMPUTATION_HEURISTICS,
-            memory_optimizer_target_node_name_regexp='(.+/)gradients/'),
+            memory_optimizer_target_node_name_prefix='optimizer/gradients/'),
         original_metagraph)
     self.assertGreater(
         len(rewritten_graph_def.node),