pytorch/xla

Pull Requests Commits

Update README Colab (#1238)

jysohn23 committed 6 years ago

Verified fb925bbe

Sllows setting floating point precision for XLA devices (#1236)

mruberry committed 6 years ago

Verified 2c25c92b

Added infrastructure to test for metrics and counter changes in tests. (#1235)

dlibenzi committed 6 years ago

6aba2fec

Remove override for _dim_arange and add override for arange_out. (#1234)

ailzhang committed 6 years ago

Verified c8b9f2b9

Added aten::upsample_nearest2d and aten::upsample_nearest2d_backward for TPU.

dlibenzi committed 6 years ago

ed0cdd84

Added GCS access utility module.

dlibenzi committed 6 years ago

79ff8634

Revert changes made to symeig.cpp

vishwakftw committed 6 years ago

9f9e3883

xla::Zeros --> xla::Zero, and special case test

vishwakftw committed 6 years ago

deb3b7e4

Remove bad code

vishwakftw committed 6 years ago

25a8773c

Modify symeig behavior when eigenvectors=False

vishwakftw committed 6 years ago

53d04e76

Logging tensorboard consistency (#1224)

zcain117 committed 6 years ago

Verified e900cfd3

Build GCS file system support.

dlibenzi committed 6 years ago

ab9162e3

Add overrides for native_layer_norm and its backward.

Ailing Zhang committed 6 years ago

d9977c6e

Let the caller handle the separate paths to be fed to the xm.save() API in case master_only is False.

dlibenzi committed 6 years ago

531a26f2

Add helper function to get Hlo graph in test.

Ailing Zhang committed 6 years ago

6bacd3ad

Added TFRecord Example reader interface.

dlibenzi committed 6 years ago

0d4c160f

Allow multi-processing POD to recognize local devices.

dlibenzi committed 6 years ago

3e62c8bb

Add missing instructions to the README

sprt committed 6 years ago

ee83d05b

Tweak README to make it more beginner-friendly

sprt committed 6 years ago

12a9b29e

Add API to wait for pending async ops. Make OpByOp use a hash key seed based on replication devices.

dlibenzi committed 6 years ago

4d2c8746

Fix clang-format cmd typo

jysohn23 committed 6 years ago

a15bd007

Log thread IDs by default.

dlibenzi committed 6 years ago

ba523346

Added mse_loss and mse_loss_backward lowering. (#1206)

ailzhang committed 6 years ago

Verified 950e12a6

Moved reduction mapping API to tensor_methods.cpp

dlibenzi committed 6 years ago

de07b24d

Vlog the client mesh retrieved from client mesh master.

dlibenzi committed 6 years ago

348676e9

Fix empty env_var crash for xla_dist

jysohn23 committed 6 years ago

7a466971

Add memory_format to `to` overrides. (#1200)

ailzhang committed 6 years ago

Verified ea7767f9

Flush stderr/stdout before exiting due to error.

jysohn23 committed 6 years ago

28581a65

Added MarkStep counter.

dlibenzi committed 6 years ago

b6c59a36

Wrapping namespace Reduction in namespace at (#1176)

yf225 committed 6 years ago

4d6f7da9

Older