Add diagonal to ATen XLA tensor

Commit

6 years ago

Add diagonal to ATen XLA tensor One of the requirements for reusing the PyTorch general einsum lowering.