pytorch/xla

Pull Requests Commits

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

dlibenzi committed 7 years ago

f401497c

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

Fix breakage due to c10 Storage API change.

dlibenzi committed 7 years ago

c35e2bde

Add aten::abs operator

vishwakftw committed 7 years ago

03560382

Migrate to using the full path for include files.

dlibenzi committed 7 years ago

97547eee

Make BuildComparisonOp more compact

Alex Şuhan committed 7 years ago

b07909a3

Move logical_element_type to XLATensor::Data

Alex Şuhan committed 7 years ago

becc590e

Don't run shape inference in ComparisonOp

Alex Şuhan committed 7 years ago

bed4412d

Add assertions in MakeTensorFromXlaLiteral

Alex Şuhan committed 7 years ago

33e8d072

Only store logical element type instead of shape

Alex Şuhan committed 7 years ago

c27c6099

Use S32 as physical representation for comparison results

Alex Şuhan committed 7 years ago

3a423708

Share comparison op implementation with translator

Alex Şuhan committed 7 years ago

6816c5d8

Add ATen XLA comparison methods against scalar

Alex Şuhan committed 7 years ago

01ca6755

Add more ATen XLA comparison methods

Alex Şuhan committed 7 years ago

e5f0eaac

Older