aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/grappler
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-05-03 14:11:13 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-04 10:25:49 -0700
commita4a9e372f6af694e91ef7aaae9f23867d0ec0fc2 (patch)
treeb5059563dcff400c88db5163df82cdc1b54fc0b2 /tensorflow/python/grappler
parent518dfea0d6d45448a360a49635fe815a28730c46 (diff)
Optimize idempotent ops, e.g., Snapshot(Snapshot(x)) => Snapshot(x)
PiperOrigin-RevId: 195308675
Diffstat (limited to 'tensorflow/python/grappler')
-rw-r--r--tensorflow/python/grappler/cluster_test.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/grappler/cluster_test.py b/tensorflow/python/grappler/cluster_test.py
index 26c6f22d34..541747867f 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), 8)
+ self.assertEqual(len(op_perfs), 4)
self.assertTrue(step_stats.dev_stats)
def testNoDetailedStats(self):
@@ -129,7 +129,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), 8)
+ self.assertEqual(len(op_perfs), 4)
self.assertTrue(step_stats.dev_stats)
def testAvailableOps(self):