pytorch/xla

Pull Requests Commits

Add max_pool2d_with_indices_backward to ATen XLA tensor

asuhan committed 6 years ago

440919d6

Add MaxPool2dIndices placeholder

asuhan committed 6 years ago

95dd9520

Fix build after recent PyTorch core changes

asuhan committed 6 years ago

35c7ae78

Drop the commit by leaving open the door for future uses.

dlibenzi committed 6 years ago

9cb517a8

Wire in ATEN XLA Python bindings and added simple end-to-end test.

dlibenzi committed 6 years ago

bf22f0fd

A few more replacements.

dlibenzi committed 6 years ago

c79bf256

Apply some de-auto refactoring.

dlibenzi committed 6 years ago

e1d73823

Fix the operator kind in NllLossBackwardOp

asuhan committed 6 years ago

ab187764

Fix build after latest PyTorch changes

asuhan committed 6 years ago

02ae0f71

Fix the XLA tensor get-xla handling, and bump up the size of conv2d bkw test.

dlibenzi committed 6 years ago

af1c99fa

Add ir::MakeNode<>() utility API to avoid casting IR Nodes into Values.

dlibenzi committed 6 years ago

f6dd3118

Rename NodeOperand to Value.

dlibenzi committed 6 years ago

9eef3b57

Adjust ATen XLA tests for cloud TPU

asuhan committed 6 years ago

da0e3585

Remove explicit ir::NodeOperand calls where possible

asuhan committed 6 years ago

f82bedcc

Add nll_loss_backward to ATen XLA tensor

asuhan committed 6 years ago

032876b9

Add nll_loss to ATen XLA tensor

asuhan committed 6 years ago

11c8465e

Add mm to ATen XLA tensor

asuhan committed 6 years ago

eef47fe2

Add transpose test

asuhan committed 6 years ago

062f5d38

Add threshold_backward to ATen XLA tensor

asuhan committed 6 years ago

68dc42ce

Add _log_softmax_backward_data to ATen XLA tensor

asuhan committed 6 years ago

420f26e9

Add thnn_conv2d_backward to ATen XLA tensor

asuhan committed 6 years ago

25399bc1

Add avg_pool2d_backward to ATen XLA tensor

asuhan committed 6 years ago

2b4bfc3a

Fix formatting in third_party/xla_client/util.h

asuhan committed 6 years ago

49946e08

Do max pool with dilation through ATen interop

asuhan committed 6 years ago

83db8fe3

Hash IR Node metadata.

dlibenzi committed 6 years ago

f4f4bad5

dlibenzi committed 6 years ago

36ed6c17

Add fast has generation from IR Node objects to be able to quickly detect graph changes.

dlibenzi committed 6 years ago

947dfafa

Create device shape out of IR Node shape.

dlibenzi committed 6 years ago

3c7dcf94

Add dereference operator to access IR Node pointer from NodeOperand, and added more testing for the IR Node replace operation.

dlibenzi committed 6 years ago

d1edc278

Added automatic conversion to IR Output for the IR NodeOperand.

dlibenzi committed 6 years ago

6a295e91

Newer Older