pytorch/xla

Pull Requests Commits

Do not rely only on the optimizer state to sync the tensors.

dlibenzi committed 6 years ago

bfe012ee

Use tf2xla helpers to implement convolution gradients

asuhan committed 6 years ago

2fcdb6cd

Update to latest TF

asuhan committed 6 years ago

a5bc2d6a

Fix build after upstream changes to TensorImpl

asuhan committed 6 years ago

132d71c7

Add xla::PRED handling in tensor utils.

dlibenzi committed 6 years ago

89b55191

Trivial comment format fix.

dlibenzi committed 6 years ago

c5b24b9c

Use the new TensorImpl constructor without storage

asuhan committed 6 years ago

1bc4d8b5

Revert "Use tf2xla helpers to implement convolution gradients"

asuhan committed 6 years ago

0b684801

Call back into the add() API instead of expanding Torch Tensor tests.

dlibenzi committed 6 years ago

330a32be

Added utility scripts to parse the XLA_SAVE_TENSORS_FILE and metrics report outputs.

dlibenzi committed 6 years ago

ba135aae

Add embedding_dense_backward to ATen XLA tensor

asuhan committed 6 years ago

f192bcb4

Use tf2xla helpers to implement convolution gradients

asuhan committed 6 years ago

4c0feb4f

Fix newly added tests tolerance on TPU

asuhan committed 6 years ago

5c20d296

Add _s_where to ATen XLA tensor

asuhan committed 6 years ago

a59187ed

Allow benchmarking executions w/out transfering the results.

dlibenzi committed 6 years ago

2cc77e96

Add bernoulli to ATen XLA tensor

asuhan committed 6 years ago

3b199812

Add kl_div_backward to ATen XLA tensor

asuhan committed 6 years ago

b8ca8185

Add kl_div to ATen XLA tensor

asuhan committed 6 years ago

208a6081

Add binary_cross_entropy_with_logits[_backward] to ATen XLA tensor

asuhan committed 6 years ago

bd811ed1

Add margin_ranking_loss to ATen XLA tensor

asuhan committed 6 years ago

9cab4715

Add triplet_margin_loss to ATen XLA tensor

asuhan committed 6 years ago

de42a691

Add hinge_embedding_loss to ATen XLA tensor

asuhan committed 6 years ago

4162375c

Add cosine_embedding_loss to ATen XLA type

asuhan committed 6 years ago

b6e6d863

Add cosine_similarity to ATen XLA type

asuhan committed 6 years ago

baf1a766

Add pairwise_distance to ATen XLA type

asuhan committed 6 years ago

c03a1382

Add instance_norm to ATen XLA type

asuhan committed 6 years ago

25dcce35

Add layer_norm to ATen XLA type

asuhan committed 6 years ago

1e205ace

Add group_norm to ATen XLA type

asuhan committed 6 years ago

a6d0520c

Emit IR node frame info in IR node ToString(), when available.

dlibenzi committed 6 years ago

f0c976dc

Added script to filter out TF log lines.

dlibenzi committed 6 years ago

41bacb51

Newer Older