tree fdb2279cce4798e5812912dfe9d9d6307fb06cd9
parent e0f353e6b93c95d44fe74f56aa3cc46826d80acc
author A. Unique TensorFlower <gardener@tensorflow.org> 1638864775 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1638865010 -0800

Fix the issue when running tpu embedding test in auto graph.

PiperOrigin-RevId: 414640929
Change-Id: I9121d215e21b7ef99fbc84d9c2c0026b5bae216c