tree 24c722bf3674eb3f969f3123eca4eac17927ff10
parent 758e021115fb95e9f1ff93e1afadba7da7f3c922
author A. Unique TensorFlower <gardener@tensorflow.org> 1639616860 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1639617204 -0800

Update tf1 verison of tpu embedding mid level api to use the new op.

PiperOrigin-RevId: 416680406
Change-Id: I79ac0276764e94e9f27071339f8a100d16e28f90