fix topk decomposer crash#41074

copybara-service[bot] wants to merge 1 commit intomainopenxla/xla:mainfrom

test_900980821openxla/xla:test_900980821Copy head branch name to clipboard