tree a3df99fbf03342457ec8b063519c1cd8376fcf7d
parent 16ae544750975296dcadd1ebb8eb52190f17ab37
author Reed Wanderman-Milne <reedwm@google.com> 1643859537 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1643859863 -0800

Rollforward 0e074ae064c441185e09b1fe4b0ee87676f10266.

The issue was that the CL checked the `dtype` attribute of the input logits, but the logits might be a list or anything else convertible to a tensor. Now the logits are explicitly converted to a tensor first.
END_PUBLIC

PiperOrigin-RevId: 426035926
Change-Id: I5df63cebe034060ed569f29544b5caf48dafe77c