tree 15b9ba20aea2efe0386ea14a0eddb88a279d9549
parent 597741a69dd74208e990633495871d3d64887b73
author Bruce Fontaine <bfontain@google.com> 1611950634 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1611950992 -0800

Add correctness test for ragged embeddings with the TPUEmbedding mid level API.

PiperOrigin-RevId: 354581897
Change-Id: Ifbbd351e6879d2ac0379520730510866e4bedf1f