pytorch/xla

Pull Requests Commits

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

Fix PRED handling on TPU.

dlibenzi committed 6 years ago

897e329b

Adjust JIT's module.cpp code to make it build again after the usual breakage.

dlibenzi committed 6 years ago

45ddd56b

Route scalar values to device data to achieve graph convergence by have scalar leaf nodes being all device data.

dlibenzi committed 6 years ago

929415ae

asuhan committed 6 years ago

af98447b

Route bitwise binary with shape checks to our implementations

asuhan committed 6 years ago

9c1f79f1

Merge pull request #564 from pytorch/fix_jit_breakage_N_plus_1

dlibenzi committed 6 years ago

Verified 7d817be8

Newer Older