diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-21 15:49:58 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 16:02:09 -0700 |
| commit | 64577d4fe51246710fbf2c9ebc17bed2febe6694 (patch) | |
| tree | efe61f2f3391699cf0650d35d7c8bff388743493 /tensorflow/contrib/distribute | |
| parent | e38dc6a60b133ce3af704c61e105f7fe5801b8fa (diff) | |
Rollback change introduced on cross_towers_ops_test by previous commit.
PiperOrigin-RevId: 214057023
Diffstat (limited to 'tensorflow/contrib/distribute')
| -rw-r--r-- | tensorflow/contrib/distribute/python/cross_tower_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/distribute/python/cross_tower_ops_test.py b/tensorflow/contrib/distribute/python/cross_tower_ops_test.py index a3e1b96a68..490371477a 100644 --- a/tensorflow/contrib/distribute/python/cross_tower_ops_test.py +++ b/tensorflow/contrib/distribute/python/cross_tower_ops_test.py @@ -114,7 +114,7 @@ class CrossTowerOpsTestBase(test.TestCase, parameterized.TestCase): self.assertEqual([v.numpy() for v in left._index.values()], list(right._index.values())) else: - with self.cached_session() as sess: + with self.test_session() as sess: self.assertEqual( sess.run(list(left._index.values())), list(right._index.values())) |