pytorch/xla

Pull Requests Commits

Add softplus to ATen XLA tensor

asuhan committed 6 years ago

64a2e4e4

Refresh autograd patch.

dlibenzi committed 6 years ago

877d2e41

Move softmax node creation to ops

asuhan committed 6 years ago

378b2044

Add softmax_backward to ATen XLA tensor

asuhan committed 6 years ago

fd0ec688

Fix data parallel to uniformily distribute over device queues.

dlibenzi committed 6 years ago

d476cf67

Remove redundant hardtanh and hardtanh_ from XLATensor

asuhan committed 6 years ago

38e9d966

Add hardtanh_ to ATen XLA tensor

asuhan committed 6 years ago

429277d2

Add hardtanh_backward to ATen XLA tensor

asuhan committed 6 years ago

baafb927

Add hardtanh to ATen XLA tensor

asuhan committed 6 years ago

7772e0b5

Cached compiled XLA computation during the tensors sync operatins.

dlibenzi committed 6 years ago

2d5a5589

Refactor lowering utils to move post-order lowering into a spearate function. Make single argument ctor explicit as well.

dlibenzi committed 6 years ago

3d488c88

Add counter for IR graph trimming and use the new XLA builder API to get the full trace.

dlibenzi committed 6 years ago

a91727bd

Add tensordot to ATen XLA tensor

asuhan committed 6 years ago

8264616b

Add leaky_relu_backward to ATen XLA tensor

asuhan committed 6 years ago

4e008bb8

Replace remaining const ref at::Scalar

asuhan committed 6 years ago

709fe190

Add elu_backward to ATen XLA tensor

asuhan committed 6 years ago

e287c7b1

Added timed execution in DEBUG=1 mode to trace performance.

dlibenzi committed 6 years ago

8e84bee6

Add linear to ATen XLA tensor

asuhan committed 6 years ago

3bb45d68

Fix the fixup script to cope with some other cases of Conda site-packages structures.

dlibenzi committed 6 years ago

a9646f81

Move the XLA tensor methods implementing the ATEN function into a new tensor_methods.cpp file.

dlibenzi committed 6 years ago

4e662dde

Replication devices should return xla:* devices. Added get-count API for it as well.

dlibenzi committed 6 years ago

93e9f8b9

Add remainder to ATen XLA tensor

asuhan committed 6 years ago

fd4e6d1c

Adjust the scalar type of cumsum, cumprod

asuhan committed 6 years ago

34075c4d

Add cumsum and cumprod to ATen XLA tensor

asuhan committed 6 years ago

52c856ee

Scale gradients CRS to inverse of number of replicas.

dlibenzi committed 6 years ago

07f06bf1

Add narrow_copy to ATen XLA tensor

asuhan committed 6 years ago

1cc7e828

Refresh the autograd PR patch.

dlibenzi committed 6 years ago

02170d6a

Fix typo leftout by previous PR.

dlibenzi committed 6 years ago

70dfa105

Add chain_matmul to ATen XLA tensor

asuhan committed 6 years ago

09a29a3d

Propagate replication devices into XRT compile instances.

dlibenzi committed 6 years ago

eedd61ae

Newer Older