tree 7631691e832bc5cfe2322593732f6678fca937bb
parent 35875e5b9621daaa65371082fa957ac3fd33c5a5
author Gaurav Jain <gjn@google.com> 1591224939 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1591225311 -0700

Add GPU support for complex types in OneHot

PiperOrigin-RevId: 314621230
Change-Id: I23bb95ff53cc2e8b16d73cbc1c22f42af6461170