pytorch/xla

Pull Requests Commits

Make the topology fetch API do not require the XRT computation client to be setup.

dlibenzi committed 7 years ago

cd8743c1

punt back to cpu for xla::dot related ops with int input (#999)

ailzhang committed 7 years ago

a98273a6

add comment about half (#998)

ailzhang committed 7 years ago

Verified 3cb6e972

better error message for arange (#1002)

ailzhang committed 7 years ago

9f6a18ee

Make sign and abs work with unsigned integers

Alex Şuhan committed 7 years ago

45507ac6

Shard datasets for distributed training. (#997)

jysohn23 committed 7 years ago

Verified 6341b5c0

add global rate and wall time to train print statement (#988)

zcain117 committed 7 years ago

Verified 582f612f

address comment

Ailing Zhang committed 7 years ago

195fd9a3

fix test; address comment

Ailing Zhang committed 7 years ago

8a73b1ad

check in call site

Ailing Zhang committed 7 years ago

a1629865

allow mean return nan for 0dim

Ailing Zhang committed 7 years ago

a8c8fff2

add test for 0dim reduction ops

Ailing Zhang committed 7 years ago

c6655607

Fix descending sort when minimum values are present

Alex Şuhan committed 7 years ago

e0eff17a

fix pow for integral types (#994)

ailzhang committed 7 years ago

Verified f7fc05ab

zcain117 committed 7 years ago

1640784e

move add_Scalar to test_utils

zcain117 committed 7 years ago

374ac858

only write if logdir is not None

zcain117 committed 7 years ago

f6cd2ef4

zcain117 committed 7 years ago

a1cc68b6

default log_dir to FLAGS.logdir

zcain117 committed 7 years ago

24a7b1ae

add percent sign

zcain117 committed 7 years ago

30889dd5

print mean accuracy

zcain117 committed 7 years ago

7f9ed874

add tensorboard support

zcain117 committed 7 years ago

660e9331

Add comment regarding tensor stride

asuhan committed 7 years ago

fc7e052d

fix split with 0dim (#991)

ailzhang committed 7 years ago

Verified 54cd2534

Fix the XLA tensors updates back from CPU tensors, in case of writeable tensors.

dlibenzi committed 7 years ago

d0e5be7b

Cleaup Kokoro CI infra bits. (#987)

jysohn23 committed 7 years ago

c469575d

Fix elementwise arithmetic with zero sized dimensions

Alex Şuhan committed 7 years ago

a1b571cb

raise error for reduction ops when numel==0 (#981)

ailzhang committed 7 years ago

Verified 61fe256f

Make --tpu flag only accept single TPU (#979)

jysohn23 committed 7 years ago

Verified 719fa926

add test_set_batch_size flag (#982)

zcain117 committed 7 years ago

Verified f1d11d56

Newer Older