tree b23a39188bd9ef84494219c152729de28faf86dc
parent 79e495c8d47f60fd31a7aa55ef726ffd155995c6
author Reed Wanderman-Milne <reedwm@google.com> 1633749885 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1633750217 -0700

Roll forward https://github.com/tensorflow/tensorflow/commit/ab0ca4bbc66a476aea305f81c69e0201b5876d0a. The internal test that it broke has been fixed.

PiperOrigin-RevId: 401913101
Change-Id: I67f095899187e38101fbb10289c5e444b0a9e8c0