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