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)
534 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))539 self.assertEqual(len(config.get_logical_device_configuration(gpus[-1])), 2)739 self.assertIsNone(config.get_logical_device_configuration(gpus[-1]))
48 name: "get_logical_device_configuration"
79 self.assertLen(config.get_logical_device_configuration(cpu_device), 3)
75 configs = tf.config.get_logical_device_configuration(physical_devices[0])
1637 def get_logical_device_configuration(self, dev): member in Context
6148 `tf.config.get_logical_device_configuration`,