Searched refs:get_visible_devices (Results 1 – 8 of 8) sorted by relevance
379 def get_visible_devices(device_type=None): function406 return context.context().get_visible_devices(device_type)
417 self.assertEqual(len(config.get_visible_devices('CPU')), 1)418 self.assertGreater(len(config.get_visible_devices('GPU')), 0)423 self.assertEqual(len(config.get_visible_devices('XLA_GPU')), 0)426 self.assertEqual(len(config.get_visible_devices('CPU')), 1)427 self.assertEqual(len(config.get_visible_devices('GPU')), 0)
36 name: "get_visible_devices"
52 name: "get_visible_devices"
1202 def get_visible_devices(self, device_type=None): member in Context
53 …st_logical_devices`, `tf.config.list_physical_devices`, `tf.config.get_visible_devices`, `tf.confi…