pytorch/xla

Pull Requests Commits

Added cast operations.

dlibenzi committed 7 years ago

f12c6eac

Use CeilOfRatio and other minor cleanups

Alex Şuhan committed 7 years ago

bf85cfa8

Add split to ATen XLA tensor

Alex Şuhan committed 7 years ago

3d1bc396

Added tests for argmin and argmax with same values.

dlibenzi committed 7 years ago

e77e0cf2

Added aten::stack operation.

dlibenzi committed 7 years ago

c6566ca4

Fix AVG pool function renaming.

dlibenzi committed 7 years ago

ccb4503a

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

dlibenzi committed 7 years ago

358e336e

Print tensor values if they do not match.

dlibenzi committed 7 years ago

3488be1e

Fix Tril kind symbol

Alex Şuhan committed 7 years ago

2588370a

Fix tensor.h formatting

Alex Şuhan committed 7 years ago

de7fcee8

Add tril to ATen XLA tensor

Alex Şuhan committed 7 years ago

3bc6cd0b

Add triu to ATen XLA tensor

Alex Şuhan committed 7 years ago

95ebf31a

Added aten::argmin and aten::argmax operations.

dlibenzi committed 7 years ago

f30f3b35

Added at::expm1 and at::log1p operators

vishwakftw committed 7 years ago

9bbc08dd

WIP: Added aten::mean operation.

dlibenzi committed 7 years ago

01f0de84

Pin PT commit ID in order to be able to work until a clear fix is in place.

dlibenzi committed 7 years ago

9d933687

Added acos, cosh, asin, asinh unary ops

vishwakftw committed 7 years ago

43c03ed1

Add permute to ATen XLA tensor

Alex Şuhan committed 7 years ago

1c631871

Added aten::where operation.

dlibenzi committed 7 years ago

f49bbb09

Add unsqueeze to ATen XLA tensor

Alex Şuhan committed 7 years ago

d3d7bf40

Added aten::gather operation.

dlibenzi committed 7 years ago

42b405ce

Consolidate some code for boring 1:1 XLA mapping ops.

dlibenzi committed 7 years ago

1c1a1c20

Don't depend on translator header in data_ops header

Alex Şuhan committed 7 years ago

7563cd1c

Allow graph parameters for reshapes target size

Alex Şuhan committed 7 years ago

39dc105b

Factor out XLA lowering in squeeze operator

Alex Şuhan committed 7 years ago

9e8c7538

Use streaming in GetCanonicalDimensionIndex error

Alex Şuhan committed 7 years ago

8ffc6d4f

Add squeeze to ATen XLA tensor

Alex Şuhan committed 7 years ago

0ea2be6c

Pin build to confirmed working PyTorch commit id

Alex Şuhan committed 7 years ago

7eb178b3

Do not die of consuminsm.

dlibenzi committed 7 years ago

31a5d89c

Check XLATensorImpl is always contiguous

Alex Şuhan committed 7 years ago

112d7e32

Older