xla
Add most of the remaining loss, norm and distance functions
#565
Merged
Overview
Commits
12
Changes
View On GitHub

Add most of the remaining loss, norm and distance functions #565

asuhan merged 12 commits into master from asuhan/more_wrapper_ops
asuhan
asuhan asuhan requested a review from dlibenzi dlibenzi 6 years ago
asuhan asuhan changed the title Add most of the remaining loss functions Add most of the remaining loss, norm and distance functions 6 years ago
asuhan
dlibenzi
dlibenzi requested changes on 2019-04-06
asuhan Add group_norm to ATen XLA type
79da693a
asuhan Add layer_norm to ATen XLA type
d2591046
asuhan Add instance_norm to ATen XLA type
80acd9eb
asuhan Add pairwise_distance to ATen XLA type
232a1d82
asuhan Add cosine_similarity to ATen XLA type
c18f6785
asuhan Add cosine_embedding_loss to ATen XLA type
52c90c1c
asuhan Add hinge_embedding_loss to ATen XLA tensor
5747726c
asuhan Add triplet_margin_loss to ATen XLA tensor
0d404f69
asuhan Add margin_ranking_loss to ATen XLA tensor
7dfbadd2
asuhan Add binary_cross_entropy_with_logits[_backward] to ATen XLA tensor
9b910113
asuhan Add kl_div to ATen XLA tensor
acd3e30b
asuhan Add kl_div_backward to ATen XLA tensor
ebd58e83
asuhan asuhan force pushed from f12da6b6 to ebd58e83 6 years ago
dlibenzi
dlibenzi approved these changes on 2019-04-08
asuhan asuhan merged b8ca8185 into master 6 years ago
asuhan asuhan deleted the asuhan/more_wrapper_ops branch 6 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum