tf.config.get_soft_device_placement