diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-05 09:33:42 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-05 09:36:53 -0800 |
commit | 83a2e03627bffa0f8cad1a9b11886aba7206a988 (patch) | |
tree | bc369cf932ca7a4decc91d11364e19e3ab29ffb9 /tensorflow/python/grappler | |
parent | 23851760b7b099214bdd4f1b88156d7ac2bdd2a2 (diff) |
Enable aggressive identity node pruning in dependency optimizer.
PiperOrigin-RevId: 184539756
Diffstat (limited to 'tensorflow/python/grappler')
-rw-r--r-- | tensorflow/python/grappler/cluster_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/grappler/cluster_test.py b/tensorflow/python/grappler/cluster_test.py index 2292b2c732..10d515a364 100644 --- a/tensorflow/python/grappler/cluster_test.py +++ b/tensorflow/python/grappler/cluster_test.py @@ -45,7 +45,7 @@ class ClusterTest(test.TestCase): op_perfs, run_time, step_stats = grappler_cluster.MeasureCosts( grappler_item) self.assertTrue(run_time > 0) - self.assertEqual(len(op_perfs), 9) + self.assertEqual(len(op_perfs), 7) self.assertTrue(step_stats.dev_stats) def testNoDetailedStats(self): @@ -125,7 +125,7 @@ class ClusterTest(test.TestCase): disable_detailed_stats=False, disable_timeline=False) as gcluster: op_perfs, run_time, step_stats = gcluster.MeasureCosts(grappler_item) self.assertTrue(run_time > 0) - self.assertEqual(len(op_perfs), 9) + self.assertEqual(len(op_perfs), 7) self.assertTrue(step_stats.dev_stats) def testAvailableOps(self): |