tensorflow: is there a way to specify XLA_GPU with tensorflow?
Check that the device is there first by examining the device names in the output from the following:

from tensorflow.python.client import device_lib
device_lib.list_local_devices()


Once you've verified that 'XLA_GPU:0' is a device on your system use something like 

with tf.device('/device:XLA_GPU:0'):


Obviously you can change the names.