xla

handle single tensor for method send_to_device_single

#5317

Merged