pytorch/xla

Pull Requests Commits

A few more replacements.

dlibenzi committed 6 years ago

4b905974

Apply some de-auto refactoring.

dlibenzi committed 6 years ago

4d9d2537

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

Use the proper output index for applying operations.

dlibenzi committed 6 years ago

1d3c4388

Turn IR operation to act on NodeOperand in order to support multi-output IR Node within XLA tensors.

dlibenzi committed 6 years ago

62d6e804

dlibenzi committed 6 years ago

cc779e86

Added multi-ops IR Node return facility.

dlibenzi committed 6 years ago

cf68d03a

Added maxpool and log softmax backward tests.

dlibenzi committed 6 years ago

7eace363

Older