未验证 提交 fdf78b86 编写于 作者: J josh11b 提交者: GitHub

Merge pull request #5561 from tensorflow/josh11b-del-is_single_tower

Drop references to `is_single_tower`.
......@@ -27,21 +27,18 @@ class GetDistributionStrategyTest(tf.test.TestCase):
"""Tests for get_distribution_strategy."""
def test_one_device_strategy_cpu(self):
ds = distribution_utils.get_distribution_strategy(0)
self.assertTrue(ds.is_single_tower)
self.assertEquals(ds.num_towers, 1)
self.assertEquals(len(ds.worker_devices), 1)
self.assertIn('CPU', ds.worker_devices[0])
def test_one_device_strategy_gpu(self):
ds = distribution_utils.get_distribution_strategy(1)
self.assertTrue(ds.is_single_tower)
self.assertEquals(ds.num_towers, 1)
self.assertEquals(len(ds.worker_devices), 1)
self.assertIn('GPU', ds.worker_devices[0])
def test_mirrored_strategy(self):
ds = distribution_utils.get_distribution_strategy(5)
self.assertFalse(ds.is_single_tower)
self.assertEquals(ds.num_towers, 5)
self.assertEquals(len(ds.worker_devices), 5)
for device in ds.worker_devices:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册