Example #1
0
 def testListDevices(self):
   # Expect at least one device.
   self.assertTrue(tfe.list_devices())
Example #2
0
 def testListDevices(self):
   # Expect at least one device.
   self.assertTrue(tfe.list_devices())
Example #3
0
 def testNumGPUs(self):
   devices = tfe.list_devices()
   self.assertEqual(len(devices) - 1, tfe.num_gpus())
Example #4
0
 def testNumGPUs(self):
   devices = tfe.list_devices()
   self.assertEqual(len(devices) - 1, tfe.num_gpus())