Add bmm to ATen XLA tensor

Commit

6 years ago

Add bmm to ATen XLA tensor Can reuse matmul, just needs additional checks. Required by the einsum lowering in PyTorch.