xla

ae61f2f6 - Revert "update tensor indexing to fail on `non_xla_tensor[xla_tensor]` (#3414)" (#3622)