tree f63a64d9b15524c03e302ac513c0bb6decf5da21
parent 04b5fed45189f6f60523116fcc842caa5fb4dd38
author Jacques Pienaar <jpienaar@google.com> 1614032327 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1614033328 -0800

Fix bug in shape inference of TensorListElementShape

The shape inference propagated without checking the result type here, but this
operation can return different element type from those of the shape.

PiperOrigin-RevId: 358902169
Change-Id: I592621a6d7533cb58f140ffb0dd37c6e62d19875