pytorch/xla

Pull Requests Commits

Set dynamic dimensions for all the reshapes.

dlibenzi committed 5 years ago

528ebd82

Bring in new TF with dynamic shapes exposed via XLA builder.

dlibenzi committed 5 years ago

365a854b

Refreshes serialization.py diff

Mike Ruberry committed 5 years ago

126d71a7

break 2 installs in one line into 2 lines.

taylanbil committed 5 years ago

d413d54b

Install tensorboardX in the build wheels script

taylanbil committed 5 years ago

ce7e9027

Set both ATEN and XLA devices.

dlibenzi committed 5 years ago

df72d4e3

Skip new `test_cat_out` test

peterbell10 committed 5 years ago

638f1cad

Re-enables fixed as_strided tests (#1481)

mruberry committed 5 years ago

Verified 9dd828cd

Fix as_strided() for scalars. (#1480)

dlibenzi committed 5 years ago

40c3aa5c

Removes redundant type promotion tests from this file

mruberry committed 5 years ago

Verified 006affd1

Skips test_memory_format_factory_like_functions_preserve (#1477)

mruberry committed 5 years ago

Verified 4a4d530a

Disables failing torch tests (#1478)

mruberry committed 5 years ago

Verified 1707b5bf

Update upsample functions accordingly. (#1422)

ailzhang committed 5 years ago

Verified deab053f

Skip a new embedding test.

Ailing Zhang committed 5 years ago

b422aa82

Remove usage of default runner user. (#1469)

jysohn23 committed 5 years ago

Verified eda5b6b3

Add Fast Neural Style Transfer link (#1471)

jysohn23 committed 5 years ago

7c5ece34

Enforce a layout for AllReduce and use Current instead of Defau… (#1468)

dlibenzi committed 5 years ago

cf3adeb7

dlibenzi committed 5 years ago

9e56738c

Exclude `nostale` label from Stalebot

jysohn23 committed 5 years ago

e800d6d6

Unify code to use std::string.

dlibenzi committed 5 years ago

035878ba

Remove one instance of constref passing.

dlibenzi committed 5 years ago

21ca94d2

Pass XlaOp by value

asuhan committed 5 years ago

85c1dee5

Add example on running on GKE with TPUs

jysohn23 committed 5 years ago

7712819c

Fix the for_each rewrite API when dealing with object cross references (cycles).

dlibenzi committed 5 years ago

79abf5e3

Make sure the real sizes are accounted when averaging reduces, to deal with dynamic shapes.

dlibenzi committed 5 years ago

88b80f8f

Install Stale Bot (#1453)

jysohn23 committed 5 years ago

d70d9d58

Fix assertEqualRel for nan. (#1457)

ailzhang committed 5 years ago

Verified ae23e743

Metrics wrapper part 2 (#1449)

zcain117 committed 5 years ago

Verified 5cb06e53

Edits, easy access to multiple styles, picture (#1454)

mruberry committed 5 years ago

Verified 4e23690a

Let script exit immediately on failure (#1452)

jysohn23 committed 5 years ago

Verified 4a33f7ce

Newer Older