tree eaa869ae1a1e3289e0ecc6653332496c1348806b
parent 9e8f335a76a36611b5dffbe3c1809b9bb0e0d2bc
author Marissa Ikonomidis <marissaw@google.com> 1623194753 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1623195024 -0700

Do not create duplicate variables on the TPU

Provide an option to only create one TPU variable for each CPU
variable that must be mirrored to the TPU. By default, a TPU
variable is created for each variable arg to the
TPUPartitionedCall. If there are multiple args that are all backed
by the same variable, the same variable is mirrored to the TPU
multiple times.

PiperOrigin-RevId: 378272842
Change-Id: I0fb80cb7afbb7a4dc841bd10530b212692d4c937