tree d9472f874db6a6ce23b05acf782ab9637edebd59
parent 4e1db954d3abb4460b86fb5f7d944e42cf611cbd
author Mehdi Amini <aminim@google.com> 1598500364 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1598500655 -0700

Re-apply 328481481 after fixing more internal issues

PiperOrigin-RevId: 328665093
Change-Id: I1111d69d96a87969013bf11890672ea369cac214