tree a627c44185f4f44f3bb89841d5f6f5f32e6b840b
parent 38168415ea5bda4c04da6d55272354274da9bc52
author Ruoxin Sang <rxsang@google.com> 1582158985 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1582159331 -0800

Make TPUStrategy work with tf.function(experimental_compile=True). This involves two changes:

1. Only create replicated var handle inside TPUReplicateContext.
2. If the function annotated with experimental_compile=True is called inside a XLAControlFlowContext, don't create a new XLAControlFlowContext.

PiperOrigin-RevId: 296086034
Change-Id: I821f3b3cd5ba69cd4c7bdb9c28e13e4b4c83f967