xla

[torch_xla2] Enable nn.functional.cosine_embedding_loss

#8368

Merged