pytorch/xla

Pull Requests Commits

Use tf2xla helpers to implement convolution gradients

Alex Şuhan committed 7 years ago

4c0feb4f

Fix newly added tests tolerance on TPU

Alex Suhan committed 7 years ago

5c20d296

Add _s_where to ATen XLA tensor

Alex Şuhan committed 7 years ago

a59187ed

Allow benchmarking executions w/out transfering the results.

dlibenzi committed 7 years ago

2cc77e96

Add bernoulli to ATen XLA tensor

Alex Şuhan committed 7 years ago

3b199812

Add kl_div_backward to ATen XLA tensor

Alex Şuhan committed 7 years ago

b8ca8185

Add kl_div to ATen XLA tensor

Alex Şuhan committed 7 years ago

208a6081

Add binary_cross_entropy_with_logits[_backward] to ATen XLA tensor

Alex Şuhan committed 7 years ago

bd811ed1

Add margin_ranking_loss to ATen XLA tensor

Alex Şuhan committed 7 years ago

9cab4715

Add triplet_margin_loss to ATen XLA tensor

Alex Şuhan committed 7 years ago

de42a691

Add hinge_embedding_loss to ATen XLA tensor

Alex Şuhan committed 7 years ago

4162375c

Add cosine_embedding_loss to ATen XLA type

Alex Şuhan committed 7 years ago

b6e6d863

Add cosine_similarity to ATen XLA type

Alex Şuhan committed 7 years ago

baf1a766

Add pairwise_distance to ATen XLA type

Alex Şuhan committed 7 years ago

c03a1382

Add instance_norm to ATen XLA type

Alex Şuhan committed 7 years ago

25dcce35

Add layer_norm to ATen XLA type

Alex Şuhan committed 7 years ago

1e205ace

Add group_norm to ATen XLA type

Alex Şuhan committed 7 years ago

a6d0520c

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

dlibenzi committed 7 years ago

f0c976dc

Added script to filter out TF log lines.

dlibenzi committed 7 years ago

41bacb51

Fix PRED handling on TPU.

dlibenzi committed 7 years ago

897e329b

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

dlibenzi committed 7 years ago

45ddd56b

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

dlibenzi committed 7 years ago

929415ae

Alex Şuhan committed 7 years ago

af98447b

Route bitwise binary with shape checks to our implementations

Alex Şuhan committed 7 years ago

9c1f79f1

Merge pull request #564 from pytorch/fix_jit_breakage_N_plus_1

dlibenzi committed 7 years ago

Verified 7d817be8

Merge pull request #561 from pytorch/lower_scalar_preds

dlibenzi committed 7 years ago

Verified b67b763d

Fix the next JIT build breakage.

dlibenzi committed 7 years ago

be8e6558

Merge pull request #560 from pytorch/index_op_cast

dlibenzi committed 7 years ago

Verified 0a6dcb30

Merge pull request #557 from pytorch/binary_ops_promoted_shape_fix

dlibenzi committed 7 years ago

Verified 85b73201

Handle xla::PRED as scalar.

dlibenzi committed 7 years ago

fa8d47e3

Newer Older