pytorch/xla

Pull Requests Commits

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

Merge pull request #561 from pytorch/lower_scalar_preds

dlibenzi committed 6 years ago

Verified b67b763d

Fix the next JIT build breakage.

dlibenzi committed 6 years ago

be8e6558

Merge pull request #560 from pytorch/index_op_cast

dlibenzi committed 6 years ago

Verified 0a6dcb30

Merge pull request #557 from pytorch/binary_ops_promoted_shape_fix

dlibenzi committed 6 years ago

Verified 85b73201

Handle xla::PRED as scalar.

dlibenzi committed 6 years ago

fa8d47e3

Cast source values into destination type for index operations.

dlibenzi committed 6 years ago

c484bef2

Fix binary ops shape inference.

dlibenzi committed 6 years ago

32606534

Merge pull request #556 from pytorch/fix_build_breakage_N

dlibenzi committed 6 years ago

Verified 49dee0ab

Fix the next build breakage WRT the JIT APIs.

dlibenzi committed 6 years ago

73606033

Fix build after PyTorch core changes

asuhan committed 6 years ago

f76ac65b

Check shapes in equality / close test functions

asuhan committed 6 years ago

37c6df4e

Fix shape of unary min / max

asuhan committed 6 years ago

50bcf7de

Fix output shape of a permutation view

asuhan committed 6 years ago

fe9941f6

Merge pull request #553 from pytorch/grab_optimizer_state_in_sync

dlibenzi committed 6 years ago

Verified 56c5e8fd

Grab the optimizer's state during sync.

dlibenzi committed 6 years ago

7cc8b15c

Newer Older