Searched refs:get_logical_device_configuration (Results 1 – 8 of 8) sorted by relevance
768 def get_logical_device_configuration(device): function802 return context.context().get_logical_device_configuration(device)
529 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))534 self.assertEqual(len(config.get_logical_device_configuration(gpus[-1])), 2)734 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))
48 name: "get_logical_device_configuration"
83 self.assertLen(config.get_logical_device_configuration(cpu_device), 3)
79 configs = tf.config.get_logical_device_configuration(physical_devices[0])
1550 def get_logical_device_configuration(self, dev): member in Context
3341 …sible_devices`, `tf.config.set_visible_devices`, `tf.config.get_logical_device_configuration`, `tf…