Enable tpu.rewrite to work on XLA CPU/GPU backends.
authorA. Unique TensorFlower <gardener@tensorflow.org>
Tue, 22 May 2018 06:37:12 +0000 (23:37 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 22 May 2018 06:40:05 +0000 (23:40 -0700)
commit028993236b2ee9674ab11294e9985d7beaf376bb
tree943ef813c7f38b0656926163217dfbc21a368a11
parenteab53f2cea0506d869b14713c6c532e0bbfd9c52
Enable tpu.rewrite to work on XLA CPU/GPU backends.

PiperOrigin-RevId: 197517946
tensorflow/compiler/tf2xla/cc/BUILD
tensorflow/contrib/tpu/ops/replication_ops.cc
tensorflow/contrib/tpu/python/tpu/tpu.py