tree 552efae886d2e6a7729f109ed7f9cfa58cc8cc7b
parent 63c706ea8492c6140e5d164c351f0559ae526106
author Sagun Bajra <sagunb@google.com> 1634058839 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1634059484 -0700

Fix failures with forwardprop_test_xla_gpu

PiperOrigin-RevId: 402593920
Change-Id: I159b8d615cbb4425bc360f203de20e35cb460900