xla
Add more tensor backward operations
#190
Merged

Add more tensor backward operations #190

asuhan merged 9 commits into master from asuhan/more_tensor_backward
asuhan
asuhan Add avg_pool2d_backward to ATen XLA tensor
d11ff422
asuhan asuhan requested a review from dlibenzi dlibenzi 6 years ago
asuhan asuhan force pushed from 80c2fe3a to 46f4ca56 6 years ago
asuhan Add thnn_conv2d_backward to ATen XLA tensor
375cc162
asuhan Add _log_softmax_backward_data to ATen XLA tensor
22c4948a
asuhan Add threshold_backward to ATen XLA tensor
f1cd73b1
asuhan Add transpose test
a372cc09
asuhan Add mm to ATen XLA tensor
8fb7d210
asuhan Add nll_loss to ATen XLA tensor
2e6f08d2
asuhan Add nll_loss_backward to ATen XLA tensor
578ebac3
asuhan asuhan force pushed from 46f4ca56 to 578ebac3 6 years ago
dlibenzi
dlibenzi requested changes on 2019-02-06
asuhan Remove explicit ir::NodeOperand calls where possible
19fd5aff
dlibenzi
dlibenzi approved these changes on 2019-02-06
asuhan asuhan merged f82bedcc into master 6 years ago
asuhan asuhan deleted the asuhan/more_tensor_backward branch 6 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone