tree fe2a0776a4faba9fab941b5ab1e224c701a8e909
parent 3256723c82b25be10bb5a3c747bc870cf005178b
author A. Unique TensorFlower <gardener@tensorflow.org> 1582880734 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1582880979 -0800

Fix reliance on known static tensor shape in `_filter_top_k`.

PiperOrigin-RevId: 297789993
Change-Id: I3877fd5decde066e320399f0b5646b74036d53c5