xla

Fix code by coping with the fact that ATEN copy() is no more.

#620

Merged