tree b18acbc5ff783e4c91829d968e7558a7ac9a5172
parent 596c05a159b6fbb9e39ca10b3f7753b7244fa1e9
author Mihai Maruseac <mihaimaruseac@google.com> 1620263222 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1620263897 -0700

Add missing validation, prevent heap OOB

PiperOrigin-RevId: 372246723
Change-Id: I1a454a643810e77d7d14821b342098c56a09fbbf