projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: c0bf28e) | patch
[XLA:GPU] Implement trivial (one-replica) cross-replica-sum on XLA:GPU.
authorJustin Lebar <jlebar@google.com>
Tue, 22 May 2018 03:41:26 +0000 (20:41 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 22 May 2018 03:43:56 +0000 (20:43 -0700)
commiteab53f2cea0506d869b14713c6c532e0bbfd9c52
tree60ef8fa706ad613261484666e66411df38d3969ftree | snapshot
parentc0bf28ecc311759ac80e12515ad931b077aae635commit | diff
[XLA:GPU] Implement trivial (one-replica) cross-replica-sum on XLA:GPU.

Also fix the CPU implementation to work in the case when there are
multiple operands to the cross-replica-sum op.

PiperOrigin-RevId: 197506311
tensorflow/compiler/xla/service/cpu/ir_emitter.cc diff | blob | history
tensorflow/compiler/xla/service/gpu/gpu_copy_insertion.cc diff | blob | history
tensorflow/compiler/xla/service/gpu/ir_emitter_unnested.cc diff | blob | history
tensorflow/compiler/xla/service/gpu/ir_emitter_unnested.h diff | blob | history
tensorflow/compiler/xla/tests/BUILD diff | blob | history
tensorflow/compiler/xla/tests/cross_replica_sum_test.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/tests/hlo_test_base.cc diff | blob | history
tensorflow/compiler/xla/tests/hlo_test_base.h diff | blob | history
Domain: Machine Learning / ML Framework;