pytorch/xla

Pull Requests Commits

Added aten::cat operation.

dlibenzi committed 6 years ago

c1bfdd82

Fix Permute::ToString

asuhan committed 6 years ago

7eae43e2

Enable the conversion API to work across allowed types.

dlibenzi committed 6 years ago

cb21840f

Make view() a static method.

dlibenzi committed 6 years ago

29f98ebf

Add text graph output.

dlibenzi committed 6 years ago

995935af

Add repeat to ATen XLA tensor

asuhan committed 6 years ago

7b65e220

Add at::atan, at::tan, at::tanh

vishwakftw committed 6 years ago

6d9bf207

Added support for exporting IR graphs in DOT format.

dlibenzi committed 6 years ago

830c6e50

Added aten::nll_loss_forward (called by Python MNIST test).

dlibenzi committed 6 years ago

f6dbb693

Made the NotSupported IR Node to have its own ID and carry a provenience description.

dlibenzi committed 6 years ago

a2eba815

Added _log_softmax() ATEN override (needed by MNIST).

dlibenzi committed 6 years ago

6024b8f7

Support post-order generation from multiple nodes.

dlibenzi committed 6 years ago

fedea843

Consolidate tensor type mapping code.

dlibenzi committed 6 years ago

e4548028

Generate conversions TPU can understand.

dlibenzi committed 6 years ago

16868b0b

Fix adaptive AVG pool and XLA where ops.

dlibenzi committed 6 years ago

13021473

Added cast operations.

dlibenzi committed 6 years ago

7356cde7

Add at::erf and at::erfc operators

vishwakftw committed 6 years ago

fc341bd0

Added aten::sigmoid operations.

dlibenzi committed 6 years ago

6a73deb0

Added aten::sum operations.

dlibenzi committed 6 years ago

64bd93a2

Use CeilOfRatio and other minor cleanups

asuhan committed 6 years ago

bf85cfa8

Add split to ATen XLA tensor

asuhan committed 6 years ago

3d1bc396

Added tests for argmin and argmax with same values.

dlibenzi committed 6 years ago

e77e0cf2

Added aten::stack operation.

dlibenzi committed 6 years ago

c6566ca4

Fix AVG pool function renaming.

dlibenzi committed 6 years ago

ccb4503a

Added aten::expand and aten::expand_as operations.

dlibenzi committed 6 years ago

358e336e

Print tensor values if they do not match.

dlibenzi committed 6 years ago

3488be1e

Fix Tril kind symbol

asuhan committed 6 years ago

2588370a

Fix tensor.h formatting

asuhan committed 6 years ago

de7fcee8

Add tril to ATen XLA tensor

asuhan committed 6 years ago

3bc6cd0b

Add triu to ATen XLA tensor

asuhan committed 6 years ago

95ebf31a

Older