Enable is_stable argument in top_k Ops (#23367)