Use stable_sort instead of sort in HloEvaluator implementation of the sort op.
We need to guarantee stable sorting when used by TopK. PiperOrigin-RevId: 222399261
Loading
Please sign in to comment
We need to guarantee stable sorting when used by TopK. PiperOrigin-RevId: 222399261