tf.config.LogicalDeviceConfiguration