tree 5e3bbf0215c902b30c5e0bd56be616375ddc575c
parent d1cac8472a2e599e3077ec96d07f1f4006c958c6
author Chenkai Kuang <chenkai@google.com> 1619024800 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1619026228 -0700

Correctly set distributed variable's update_uid if initialized from a checkpoint using `CheckpointInitialValueCallable`.

PiperOrigin-RevId: 369679126
Change-Id: Ib5c449dfdf02a95f804ad6132359017d3b38d49e