pytorch/xla

Pull Requests Commits

Code format and flush while printing test/train updates. (#1594)

taylanbil committed 5 years ago

Verified 67fe36dd

Single Core Alexnet on Fashion MNIST Colab

Mike Ruberry committed 5 years ago

d02f2789

Standardize further to use XLA constants. (#1589)

dlibenzi committed 5 years ago

Verified 5bc5a0fd

Remove noop convert ops.

dlibenzi committed 5 years ago

d9f804a4

Use multiplication in BF16 test.

dlibenzi committed 5 years ago

62ea1483

Explicitly set dtype for arange(). (#1581)

dlibenzi committed 5 years ago

Verified 71bcbbd5

Use XLA constants functions. (#1578)

dlibenzi committed 5 years ago

Verified b2743833

Minor BN refactor. (#1579)

dlibenzi committed 5 years ago

Verified 52d6a6fc

Avoid missing DISPLAY environment error when running on termina… (#1575)

dlibenzi committed 5 years ago

Verified 2463ede4

Adds README preamble highlighting Colabs and README sections

Mike Ruberry committed 5 years ago

75d3bc56

Add more bfloat16 mixed precision tests.

dlibenzi committed 5 years ago

3b1896c0

Recover from TPU Maintenance Events (#1568)

jysohn23 committed 5 years ago

Verified 101ef788

Throw if empty() gets negative dimensions.

dlibenzi committed 5 years ago

a7ffd87c

Disable TestNNDeviceType.test_nll_loss_empty_tensor_reduction_mean because of division 0 by 0

pbelevich committed 5 years ago

75dfeb5d

Add missing ClusterResolver import

Myle Ott committed 5 years ago

d61dc40b

Replace tensorflow::gtl::ArraySlice with absl::Span.

Ailing Zhang committed 5 years ago

5fdf8dcd

Integrate metrics coming from the TF/XRT service side.

dlibenzi committed 5 years ago

ea8e0b49

Fix to print out the max accuracy correctly

monologg committed 5 years ago

362bfb3e

Bring in new TF.

dlibenzi committed 5 years ago

c129383a

Fix comment about Bazel version.

dlibenzi committed 5 years ago

13e01518

Bring in new TF with XRT metrics.

dlibenzi committed 5 years ago

8c6decc2

Remove use of ldconfig during builds

jysohn23 committed 5 years ago

0926885d

Adds "Getting Started" Colab notebook (#1553)

mruberry committed 5 years ago

Verified 3cb90954

Do not depend anymore from pytorch common_utils and use directly Python unittest.

dlibenzi committed 5 years ago

4270d41b

Record XLA metrics in Tensorboard Summary (#1551)

zcain117 committed 5 years ago

Verified 0a94c329

pritamdamania committed 5 years ago

fce6901e

Fix X10-device_test.diff for https://github.com/pytorch/pytorch/pull/30445

pritamdamania committed 5 years ago

469e2443

Fix X10-device_test.diff for https://github.com/pytorch/pytorch/pull/30445

pritamdamania committed 5 years ago

99b55ce7

Fix X10-device_test.diff for https://github.com/pytorch/pytorch/pull/30445

pritamdamania committed 5 years ago

77a9afee

Refactor `xla_dist.py`: breakout `worker.py` and `cluster.py` (#1549)

jysohn23 committed 5 years ago

Verified ff80898a

Newer Older