xla

9199f36f - Send tensors to XLA device before running tests, in case of tensor-modifying test functions.