xla

5b1f27c3 - Added ability to select the reference parameter depending on the ATEN function signature.

Added ability to select the reference parameter depending on the ATEN function signature. This, together with a small PT PR to be sent, fixes the CPU transfer isue and allow printing and comparing tensors from the Python side.