pytorch/xla

Pull Requests Commits

Allow test filtering via -F option.

dlibenzi committed 7 years ago

40bdc83b

Make Node::ReturnOp(s) check the number of outputs

Alex Şuhan committed 7 years ago

eba59e27

Fix typo: GetTestTesor -> GetTestTensor

Alex Şuhan committed 7 years ago

bf00c4c6

Add batch_norm to ATen XLA tensor

Alex Şuhan committed 7 years ago

013b090f

Update to latest TF

Alex Şuhan committed 7 years ago

2c7de4c3

Move hasher implementation to CPP file to reuse the hash combiner, instead of copying its code.

dlibenzi committed 7 years ago

b062728b

Drop explicit initialization.

dlibenzi committed 7 years ago

d9189335

Added PT select() support in XLA.

dlibenzi committed 7 years ago

b7e688cc

Use dereference operator in IR Values.

dlibenzi committed 7 years ago

c193320e

Added sub* operations, and zero_.

dlibenzi committed 7 years ago

ab8045e5

Add some VLOG output from hooks.

dlibenzi committed 7 years ago

3a3ba533

Implement numel() on the XLA side.

dlibenzi committed 7 years ago

7c1044b3

Add layout patch.

dlibenzi committed 7 years ago

19b69f42

The _like() operations needs unwrap as the autograd module calls them with variable inputs.

dlibenzi committed 7 years ago

9a6272c4

Move API to anon namespace.

dlibenzi committed 7 years ago

728069c7

Use ignore-in-shared-libs to ignore symbols only from shared libraries.

dlibenzi committed 7 years ago

cb051ca8

Minor code refactoring to not use the object shape as input, and comment also fixed from previously copied one.

dlibenzi committed 7 years ago

87619676

Directly implement the _s_copy_from() API.

dlibenzi committed 7 years ago

188797f5

Added the missing bare ones() and zeros().

dlibenzi committed 7 years ago

714b577d

Add pending PT PRs.

dlibenzi committed 7 years ago

1232dc21

dlibenzi committed 7 years ago

52bdca20

Added ability to select the reference parameter depending on the ATEN function signature.

dlibenzi committed 7 years ago

5b1f27c3

Allow for device-generating APIs to return a device optionally, so that the XLA tensor creationg API in the bridge can pass the input tensors straight through.

dlibenzi committed 7 years ago

67825882

Allow for ATEN full signature match during black-listing.

dlibenzi committed 7 years ago

d3bd1f00

Prefer to create constants instead of device data.

dlibenzi committed 7 years ago

6cb84d65

Lower ATEN ones and zeros to XLA broadcasted scalar ops.

dlibenzi committed 7 years ago

d7dcacac

Unwrap a variable to get the raw ATEN tensor data.

dlibenzi committed 7 years ago

6d618060

If a tensor is already CPU, should be passed as is out of the conversion API.

dlibenzi committed 7 years ago

5fe139a5

Instead of implementing constructor like functions in the bridge, simply rewrite the options to cast to CPU device.

dlibenzi committed 7 years ago

a1fae394

Add thnn_conv2d_forward to ATen XLA tensor

Alex Şuhan committed 7 years ago

cfa4f432

Older