xla

d4c90c91 - [torch_xla2] Enable nn.functional.cosine_embedding_loss (#8368)